From b31daadc034cbaa8838d78016e9ec4a9af3a1d5c Mon Sep 17 00:00:00 2001
From: comfyanonymous <comfyanonymous@protonmail.com>
Date: Tue, 28 Feb 2023 12:27:43 -0500
Subject: [PATCH] Try to improve memory issues with del.

---
 comfy/sd.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/comfy/sd.py b/comfy/sd.py
index 0b55b8b1..4ac11eba 100644
--- 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):