From 668cef32e4bc566e2db1ca2a99bfc935825cff9c Mon Sep 17 00:00:00 2001 From: lllyasviel Date: Thu, 25 Jan 2024 12:18:08 -0800 Subject: [PATCH] Update forge_loader.py --- modules_forge/forge_loader.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/modules_forge/forge_loader.py b/modules_forge/forge_loader.py index 08162b74..6fccf7ff 100644 --- a/modules_forge/forge_loader.py +++ b/modules_forge/forge_loader.py @@ -40,6 +40,10 @@ class ForgeSD: self.clip = clip self.vae = vae self.clipvision = clipvision + self.unet_original = unet + self.clip_original = clip + self.vae_original = vae + self.clipvision_original = clipvision @contextlib.contextmanager @@ -235,10 +239,6 @@ def load_model_for_a1111(timer, checkpoint_info=None, state_dict=None): sd_model.decode_first_stage = patched_decode_first_stage sd_model.encode_first_stage = patched_encode_first_stage - sd_model.forge_objects.unet_original = forge_objects.unet - sd_model.forge_objects.clip_original = forge_objects.clip - sd_model.forge_objects.vae_original = forge_objects.vae - sd_model.forge_objects.clipvision_original = forge_objects.clipvision sd_model.clip = sd_model.cond_stage_model timer.record("forge finalize")