From 66d42332101107d66a9dc8e18d781ec49991cce8 Mon Sep 17 00:00:00 2001 From: comfyanonymous Date: Thu, 8 Aug 2024 15:16:51 -0400 Subject: [PATCH] Fix. --- comfy/model_management.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/comfy/model_management.py b/comfy/model_management.py index 23226cbefb7..b7aff9f5ef5 100644 --- a/comfy/model_management.py +++ b/comfy/model_management.py @@ -413,7 +413,7 @@ def free_memory(memory_required, device, keep_loaded=[]): break memory_to_free = memory_required - free_mem logging.debug(f"Unloading {current_loaded_models[i].model.model.__class__.__name__}") - if current_loaded_models[i].model_unload(memory_to_free, free_mem): + if current_loaded_models[i].model_unload(memory_to_free): unloaded_model.append(i) for i in sorted(unloaded_model, reverse=True):