do not stop working on failed hypernetwork load

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