Fix model merge bug.

Unload models before getting weights for model patching.
jn-jairo · Nov 10, 2023 · 3e0033e · 3e0033e
1 parent 002aefa
commit 3e0033e
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/comfy/model_patcher.py b/comfy/model_patcher.py
@@ -134,6 +134,7 @@ def add_patches(self, patches, strength_patch=1.0, strength_model=1.0):
         return list(p)
 
     def get_key_patches(self, filter_prefix=None):
+        comfy.model_management.unload_model_clones(self)
         model_sd = self.model_state_dict()
         p = {}
         for k in model_sd: