mirror of
https://github.com/lllyasviel/stable-diffusion-webui-forge.git
synced 2026-04-30 19:21:21 +00:00
Update forge_loader.py
This commit is contained in:
@@ -141,6 +141,7 @@ def load_model_for_a1111(timer, checkpoint_info=None, state_dict=None):
|
|||||||
|
|
||||||
with sd_disable_initialization.DisableInitialization(disable_clip=True):
|
with sd_disable_initialization.DisableInitialization(disable_clip=True):
|
||||||
sd_model = instantiate_from_config(a1111_config.model)
|
sd_model = instantiate_from_config(a1111_config.model)
|
||||||
|
_ = sd_model.load_state_dict(state_dict, strict=False)
|
||||||
|
|
||||||
timer.record("forge instantiate config")
|
timer.record("forge instantiate config")
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user