Skip to content

Commit

Permalink
Fix case where model was not properly unloaded in merging workflows.
Browse files Browse the repository at this point in the history
  • Loading branch information
comfyanonymous committed Aug 27, 2024
1 parent 6bbdcd2 commit 38c22e6
Showing 1 changed file with 2 additions and 0 deletions.
2 changes: 2 additions & 0 deletions comfy/model_management.py
Original file line number Diff line number Diff line change
Expand Up @@ -405,6 +405,8 @@ def unload_model_clones(model, unload_weights_only=True, force_unload=True):
if not force_unload:
if unload_weights_only and unload_weight == False:
return None
else:
unload_weight = True

for i in to_unload:
logging.debug("unload clone {} {}".format(i, unload_weight))
Expand Down

0 comments on commit 38c22e6

Please sign in to comment.