lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-03-14 17:37:25 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update forge_loader.py

Browse Source
This commit is contained in:
lllyasviel
2024-01-25 06:27:16 -08:00
parent 1d8275e982
commit 9c1cc9b577
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
modules_forge/forge_loader.py
View File

@@ -123,8 +123,8 @@ def load_checkpoint_guess_config(sd, output_vae=True, output_clip=True, output_c
def load_model_for_a1111(timer, checkpoint_info=None, state_dict=None):
a1111_config = find_checkpoint_config(state_dict, checkpoint_info)
a1111_config = OmegaConf.load(a1111_config)
a1111_config_filename = find_checkpoint_config(state_dict, checkpoint_info)
a1111_config = OmegaConf.load(a1111_config_filename)
timer.record("forge solving config")
if hasattr(a1111_config.model.params, 'network_config'):
@@ -138,6 +138,7 @@ def load_model_for_a1111(timer, checkpoint_info=None, state_dict=None):
with no_clip():
sd_model = instantiate_from_config(a1111_config.model)
sd_model.used_config = a1111_config_filename
timer.record("forge instantiate config")