do not stop working on failed hypernetwork load
This commit is contained in:
parent
d15b3ec001
commit
97bc0b9504
|
@ -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
|
||||
|
||||
|
|
Loading…
Reference in New Issue