Commit 97bc0b95 authored by AUTOMATIC's avatar AUTOMATIC
Browse files

do not stop working on failed hypernetwork load

parent d15b3ec0
Loading
Loading
Loading
Loading
+9 −2
Original line number Diff line number Diff line
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):
        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