298977887/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-01-11 02:15:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge 1f6ab7dbfb into 2ff3104f70

Browse Source
This commit is contained in:
Alexander Piskun 2025-01-10 11:39:41 -05:00 committed by GitHub
commit acf81c0c9d
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
main.py
View File

@ -166,6 +166,14 @@ def prompt_worker(q, server_instance):
queue_item = q.get(timeout=timeout) queue_item = q.get(timeout=timeout)
if queue_item is not None: if queue_item is not None:
item, item_id = queue_item item, item_id = queue_item
if item[3].get("unload_models"):
# For those cases where the flag is set, to clear memory before execution
comfy.model_management.unload_all_models()
gc.collect()
comfy.model_management.soft_empty_cache()
last_gc_collect = time.perf_counter()
execution_start_time = time.perf_counter() execution_start_time = time.perf_counter()
prompt_id = item[1] prompt_id = item[1]
server_instance.last_prompt_id = prompt_id server_instance.last_prompt_id = prompt_id