Skip to content

Commit 38c22e6

Browse files
Fix case where model was not properly unloaded in merging workflows.
1 parent 6bbdcd2 commit 38c22e6

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

comfy/model_management.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -405,6 +405,8 @@ def unload_model_clones(model, unload_weights_only=True, force_unload=True):
405405
if not force_unload:
406406
if unload_weights_only and unload_weight == False:
407407
return None
408+
else:
409+
unload_weight = True
408410

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

0 commit comments

Comments
 (0)