Del the right object when applying lora.

12a6e931 · comfyanonymous · 85a8900a · 12a6e931
Commit 12a6e931 authored Jul 22, 2023 by comfyanonymous
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

comfy/sd.py comfy/sd.py +1 -1

No files found.
--- a/comfy/sd.py
+++ b/comfy/sd.py
@@ -353,7 +353,7 @@ class ModelPatcher:
            temp_weight = weight.to(torch.float32, copy=True)
            out_weight = self.calculate_weight(self.patches[key], temp_weight, key).to(weight.dtype)
            set_attr(self.model, key, out_weight)
-            del temp_weight
+            del weight
        return self.model

    def calculate_weight(self, patches, weight, key):