Fix model merge bug.

Unload models before getting weights for model patching.
1 year ago · 3e0033ef30
parent 002aefa382
commit 3e0033ef30
1 changed files with 1 additions and 0 deletions
--- a/comfy/model_patcher.py
+++ b/comfy/model_patcher.py
@ -134,6 +134,7 @@ class ModelPatcher:
        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: