Fix.

NeoWorldTeam · Aug 8, 2024 · 66d4233 · 66d4233
1 parent 591010b
commit 66d4233
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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):