298977887/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-07-20 22:47:06 +08:00
Code Issues Packages Projects Releases Wiki Activity

Run garbage collector when switching workflow if needed.

Browse Source
This commit is contained in:
comfyanonymous 2024-11-01 20:25:38 -04:00
parent bd5d8f150f
commit 5c106a0308
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
execution.py
View File

@ -480,6 +480,7 @@ class PromptExecutor:
if self.caches.outputs.get(node_id) is not None: if self.caches.outputs.get(node_id) is not None:
cached_nodes.append(node_id) cached_nodes.append(node_id)
comfy.model_management.cleanup_models_gc()
self.add_message("execution_cached", self.add_message("execution_cached",
{ "nodes": cached_nodes, "prompt_id": prompt_id}, { "nodes": cached_nodes, "prompt_id": prompt_id},
broadcast=False) broadcast=False)