Move cleanup_models to improve performance.

2024-03-23 17:27:10 -04:00
parent a28a9dc836
commit 6a32c06f06
2 changed files with 1 additions and 1 deletions
--- 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