From eff32778a6a459dc0055cd3971e49ea598689332 Mon Sep 17 00:00:00 2001
From: Ash Martian <ash.lee.martian@gmail.com>
Date: Sun, 15 Dec 2024 18:53:57 -0800
Subject: [PATCH] Don't keep models loaded on cleanup

- **Exception Message:** cleanup_models() got an unexpected keyword argument 'keep_clone_weights_loaded'
---
 workflow.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/workflow.py b/workflow.py
index cbe54a6..881843f 100644
--- a/workflow.py
+++ b/workflow.py
@@ -486,7 +486,7 @@ def execute(server, prompt, prompt_id, extra_data={}, execute_outputs=[]):
             if caches.outputs.get(node_id) is not None:
                 cached_nodes.append(node_id)
 
-        comfy.model_management.cleanup_models(keep_clone_weights_loaded=True)
+        comfy.model_management.cleanup_models()
         add_message(server, "execution_cached",{"nodes": cached_nodes, "prompt_id": prompt_id}, broadcast=False)
         pending_subgraph_results = {}
         executed = set()