Move cleanup_models to improve performance.

6a32c06f · comfyanonymous · a28a9dc8 · 6a32c06f · 6a32c06f
Commit 6a32c06f authored Mar 23, 2024 by comfyanonymous
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

execution.py execution.py +0 -1

main.py main.py +1 -0

No files found.
--- a/execution.py
+++ b/execution.py
@@ -368,7 +368,6 @@ class PromptExecutor:
                    d = self.outputs_ui.pop(x)
                    del d
-            comfy.model_management.cleanup_models()
            self.add_message("execution_cached",
                          { "nodes": list(current_outputs) , "prompt_id": prompt_id},
                          broadcast=False)

--- a/main.py
+++ b/main.py
@@ -139,6 +139,7 @@ def prompt_worker(q, server):
        if need_gc:
            current_time = time.perf_counter()
            if (current_time - last_gc_collect) > gc_collect_interval:
+                comfy.model_management.cleanup_models()
                gc.collect()
                comfy.model_management.soft_empty_cache()
                last_gc_collect = current_time