From 9f08eaa09a64e6087a308d5491d0ae3cb882c651 Mon Sep 17 00:00:00 2001 From: lllyasviel Date: Thu, 25 Jan 2024 05:20:42 -0800 Subject: [PATCH] Update forge_loader.py --- modules_forge/forge_loader.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/modules_forge/forge_loader.py b/modules_forge/forge_loader.py index 7ae02258..291a4910 100644 --- a/modules_forge/forge_loader.py +++ b/modules_forge/forge_loader.py @@ -206,6 +206,8 @@ def load_model_for_a1111(timer, checkpoint_info=None, state_dict=None): timer.record("forge finalize") sd_model.unet_patcher = forge_object.unet + sd_model.clip_patcher = forge_object.clip.patcher + sd_model.vae_patcher = forge_object.vae.patcher return sd_model