lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-04-28 18:21:48 +00:00
Code Issues Packages Projects Releases Wiki Activity

use safer codes

Browse Source
This commit is contained in:
layerdiffusion
2024-08-31 10:55:19 -07:00
parent 1f91b35a43
commit 70a555906a
3 changed files with 7 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
backend/loader.py
View File

@@ -264,7 +264,7 @@ def split_state_dict(sd, additional_state_dicts: list = None):
return state_dict, guess
@torch.no_grad()
@torch.inference_mode()
def forge_loader(sd, additional_state_dicts=None):
try:
state_dicts, estimated_config = split_state_dict(sd, additional_state_dicts=additional_state_dicts)