diff options
Diffstat (limited to 'modules')
| -rw-r--r-- | modules/hypernetwork.py | 11 |
1 files changed, 9 insertions, 2 deletions
diff --git a/modules/hypernetwork.py b/modules/hypernetwork.py index 9ed1eed9..c5cf4afa 100644 --- a/modules/hypernetwork.py +++ b/modules/hypernetwork.py @@ -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
|