do not stop working on failed hypernetwork load

This commit is contained in:
AUTOMATIC 2022-10-07 13:22:50 +03:00
parent d15b3ec001
commit 97bc0b9504
1 changed files with 9 additions and 2 deletions

View File

@ -1,5 +1,8 @@
import glob
import os
import sys
import traceback
import torch
from modules import devices
@ -36,8 +39,12 @@ def load_hypernetworks(path):
res = {}
for filename in glob.iglob(path + '**/*.pt', recursive=True):
hn = Hypernetwork(filename)
res[hn.name] = hn
try:
hn = Hypernetwork(filename)
res[hn.name] = hn
except Exception:
print(f"Error loading hypernetwork {filename}", file=sys.stderr)
print(traceback.format_exc(), file=sys.stderr)
return res