diff --git a/modules_forge/forge_loader.py b/modules_forge/forge_loader.py index b06a58a7..51413739 100644 --- a/modules_forge/forge_loader.py +++ b/modules_forge/forge_loader.py @@ -43,7 +43,12 @@ class ForgeSD: self.clipvision = clipvision def shallow_copy(self): - return ForgeSD(self.unet, self.clip, self.vae, self.clipvision) + return ForgeSD( + self.unet.clone() if self.unet is not None else None, + self.clip.clone() if self.clip is not None else None, + self.vae, + self.clipvision + ) @contextlib.contextmanager