xinyun/ComfyUI-Manager
1
0
Fork 0
mirror of https://git.datalinker.icu/ltdrdata/ComfyUI-Manager synced 2025-12-16 01:25:03 +08:00
Code Issues Packages Projects Releases Wiki Activity

[fix] Ensure batch history is written when queue becomes empty

Browse Source
This commit is contained in:
bymyself 2025-06-21 16:01:25 -07:00
parent d1ca6288a3
commit 77133375ad
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
comfyui_manager/glob/manager_server.py
View File

@ -200,7 +200,7 @@ class TaskQueue:
"""Start the task worker if not already running. Returns True if started, False if already running.""" """Start the task worker if not already running. Returns True if started, False if already running."""
if self._worker_task is not None and self._worker_task.is_alive(): if self._worker_task is not None and self._worker_task.is_alive():
logging.debug("[ComfyUI-Manager] Worker already running, skipping start") logging.debug("[ComfyUI-Manager] Worker already running, skipping start")
return False # Already running return False
logging.debug("[ComfyUI-Manager] Starting task worker thread") logging.debug("[ComfyUI-Manager] Starting task worker thread")
self._worker_task = threading.Thread(target=lambda: asyncio.run(task_worker())) self._worker_task = threading.Thread(target=lambda: asyncio.run(task_worker()))
@ -1041,7 +1041,7 @@ async def task_worker():
"[ComfyUI-Manager] Finalizing batch history with %d completed tasks", "[ComfyUI-Manager] Finalizing batch history with %d completed tasks",
task_queue.done_count(), task_queue.done_count(),
) )
task_queue.finalize() await task_queue.finalize()
logging.debug("[ComfyUI-Manager] Batch finalization complete") logging.debug("[ComfyUI-Manager] Batch finalization complete")
logging.info("\nAfter restarting ComfyUI, please refresh the browser.") logging.info("\nAfter restarting ComfyUI, please refresh the browser.")
@ -1052,8 +1052,8 @@ async def task_worker():
logging.debug("[ComfyUI-Manager] Broadcasting queue all-done status") logging.debug("[ComfyUI-Manager] Broadcasting queue all-done status")
PromptServer.instance.send_sync("cm-queue-status", res) PromptServer.instance.send_sync("cm-queue-status", res)
logging.debug("[ComfyUI-Manager] Task worker exiting") logging.debug("[ComfyUI-Manager] Task worker exiting")
return return
item, task_index = task item, task_index = task
kind = item.kind kind = item.kind