From 77133375adff5677c277f18dcc8aedc8784ecb69 Mon Sep 17 00:00:00 2001 From: bymyself Date: Sat, 21 Jun 2025 16:01:25 -0700 Subject: [PATCH] [fix] Ensure batch history is written when queue becomes empty --- comfyui_manager/glob/manager_server.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/comfyui_manager/glob/manager_server.py b/comfyui_manager/glob/manager_server.py index 799cb953..00fb067a 100644 --- a/comfyui_manager/glob/manager_server.py +++ b/comfyui_manager/glob/manager_server.py @@ -200,7 +200,7 @@ class TaskQueue: """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(): logging.debug("[ComfyUI-Manager] Worker already running, skipping start") - return False # Already running + return False logging.debug("[ComfyUI-Manager] Starting task worker thread") 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", task_queue.done_count(), ) - task_queue.finalize() + await task_queue.finalize() logging.debug("[ComfyUI-Manager] Batch finalization complete") 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") PromptServer.instance.send_sync("cm-queue-status", res) - logging.debug("[ComfyUI-Manager] Task worker exiting") - return + logging.debug("[ComfyUI-Manager] Task worker exiting") + return item, task_index = task kind = item.kind