Update forge_loader.py

This commit is contained in:
lllyasviel 2024-01-25 09:52:36 -08:00
parent 31f252d92d
commit 44c8a83c4a

View File

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