safe copy

2026-01-26 19:09:45 +00:00 · 2024-02-03 18:17:28 -08:00
parent a544ececfe
commit 8d9fdb20c3
1 changed files with 6 additions and 1 deletions
--- 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