From 38c22e631ad090a4841e4a0f015a30c565a9f7fc Mon Sep 17 00:00:00 2001 From: comfyanonymous Date: Tue, 27 Aug 2024 18:46:55 -0400 Subject: [PATCH] Fix case where model was not properly unloaded in merging workflows. --- comfy/model_management.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/comfy/model_management.py b/comfy/model_management.py index 91e692ba..4147989e 100644 --- a/comfy/model_management.py +++ b/comfy/model_management.py @@ -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))