Update forge_loader.py
This commit is contained in:
parent
31f252d92d
commit
44c8a83c4a
@ -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")
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user