From 9c1cc9b5771ad3e07bafe5ad222dc83382994326 Mon Sep 17 00:00:00 2001 From: lllyasviel Date: Thu, 25 Jan 2024 06:27:16 -0800 Subject: [PATCH] Update forge_loader.py --- modules_forge/forge_loader.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/modules_forge/forge_loader.py b/modules_forge/forge_loader.py index 54f01954..5ad00586 100644 --- a/modules_forge/forge_loader.py +++ b/modules_forge/forge_loader.py @@ -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")