Try to improve memory issues with del.

2023-02-28 12:27:43 -05:00 · 2023-02-28 12:27:43 -05:00 · b31daadc03
parent cd85f876f2
commit b31daadc03
1 changed files with 4 additions and 1 deletions
--- a/comfy/sd.py
+++ b/comfy/sd.py
@ -222,8 +222,11 @@ class ModelPatcher:
        return self.model
    def unpatch_model(self):
        model_sd = self.model.state_dict()
-        for k in self.backup:
+        keys = list(self.backup.keys())
+        for k in keys:
            model_sd[k][:] = self.backup[k]
+            del self.backup[k]
+
        self.backup = {}

 def load_lora_for_models(model, clip, lora_path, strength_model, strength_clip):