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

View File

@ -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))