From 8954f95c82ed440e9814d5c7162f080be002658c Mon Sep 17 00:00:00 2001 From: lllyasviel Date: Thu, 25 Jan 2024 14:53:56 -0800 Subject: [PATCH] Update forge_loader.py --- modules_forge/forge_loader.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/modules_forge/forge_loader.py b/modules_forge/forge_loader.py index f4b4e479..e69b3321 100644 --- a/modules_forge/forge_loader.py +++ b/modules_forge/forge_loader.py @@ -40,10 +40,9 @@ 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 + + def shallow_copy(self): + return ForgeSD(self.unet, self.clip, self.vae, self.clipvision) @contextlib.contextmanager @@ -153,6 +152,8 @@ def load_model_for_a1111(timer, checkpoint_info=None, state_dict=None): output_model=True ) sd_model.forge_objects = forge_objects + sd_model.forge_objects_original = forge_objects.shallow_copy() + sd_model.forge_objects_after_applying_lora = forge_objects.shallow_copy() timer.record("forge load real models") sd_model.first_stage_model = forge_objects.vae.first_stage_model