Skip to content

GitLab

Commit 3e0033ef authored by comfyanonymous's avatar comfyanonymous
Browse files

Fix model merge bug. 

Unload models before getting weights for model patching.
parent 002aefa3
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
comfy/model_patcher.py
View file @ 3e0033ef
...@@ -134,6 +134,7 @@ class ModelPatcher: ...@@ -134,6 +134,7 @@ class ModelPatcher:
return list(p) return list(p)
def get_key_patches(self, filter_prefix=None): def get_key_patches(self, filter_prefix=None):
comfy.model_management.unload_model_clones(self)
model_sd = self.model_state_dict() model_sd = self.model_state_dict()
p = {} p = {}
for k in model_sd: for k in model_sd:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment