lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-02-05 07:29:57 +00:00
Code Issues Packages Projects Releases Wiki Activity

better print

Browse Source
This commit is contained in:
layerdiffusion
2024-08-11 22:56:51 -07:00
parent d420522e2b
commit 2d17e8df8d
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
backend/loader.py
View File

@@ -177,7 +177,11 @@ def split_state_dict(sd, sd_vae=None):
@torch.no_grad()
def forge_loader(sd, sd_vae=None):
state_dicts, estimated_config = split_state_dict(sd, sd_vae=sd_vae)
try:
state_dicts, estimated_config = split_state_dict(sd, sd_vae=sd_vae)
except:
raise ValueError('Failed to recognize model type!')
repo_name = estimated_config.huggingface_repo
local_path = os.path.join(dir_path, 'huggingface', repo_name)