qnloft
/
ComfyUI
mirror of https://github.com/comfyanonymous/ComfyUI.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fix case where model was not properly unloaded in merging workflows.

This commit is contained in:
comfyanonymous 2024-08-27 18:46:55 -04:00
parent 6bbdcd28ae
commit 38c22e631a
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
comfy/model_management.py
View File

@ -405,6 +405,8 @@ def unload_model_clones(model, unload_weights_only=True, force_unload=True):
if not force_unload: if not force_unload:
if unload_weights_only and unload_weight == False: if unload_weights_only and unload_weight == False:
return None return None
else:
unload_weight = True
for i in to_unload: for i in to_unload:
logging.debug("unload clone {} {}".format(i, unload_weight)) logging.debug("unload clone {} {}".format(i, unload_weight))