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")