Merge pull request #12 from AshMartian/patch-1

Don't keep clone weights loaded on cleanup
2025-03-17 13:00:43 +01:00
parent d2e744eb3a eff32778a6
commit 5c3e348796
1 changed files with 1 additions and 1 deletions
@@ -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()