Fix model merge bug.

Unload models before getting weights for model patching.

Fix model merge bug.
Unload models before getting weights for model patching.
3e0033ef · comfyanonymous · 002aefa3 · 3e0033ef
Commit 3e0033ef authored Nov 10, 2023 by comfyanonymous
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

comfy/model_patcher.py comfy/model_patcher.py +1 -0

No files found.
--- 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: