openvinotoolkit/stable-diffusion-webui
1
1
Fork 0
mirror of https://github.com/openvinotoolkit/stable-diffusion-webui.git synced 2024-12-14 22:53:25 +03:00
Code Issues Packages Projects Releases Wiki Activity

do not stop working on failed hypernetwork load

Browse Source
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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
modules/hypernetwork.py
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