From 44c8a83c4a219057b742de73f8c54e8846ab5021 Mon Sep 17 00:00:00 2001 From: lllyasviel Date: Thu, 25 Jan 2024 09:52:36 -0800 Subject: [PATCH] Update forge_loader.py --- modules_forge/forge_loader.py | 1 + 1 file changed, 1 insertion(+) diff --git a/modules_forge/forge_loader.py b/modules_forge/forge_loader.py index f455a50f..4e3ede10 100644 --- a/modules_forge/forge_loader.py +++ b/modules_forge/forge_loader.py @@ -141,6 +141,7 @@ def load_model_for_a1111(timer, checkpoint_info=None, state_dict=None): with sd_disable_initialization.DisableInitialization(disable_clip=True): sd_model = instantiate_from_config(a1111_config.model) + _ = sd_model.load_state_dict(state_dict, strict=False) timer.record("forge instantiate config")