Del the right object when applying lora.

2 years ago · 12a6e93171
parent 85a8900a14
commit 12a6e93171
1 changed files with 1 additions and 1 deletions
--- 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):