From 1d63b216433c808ae206f698a7b9f8e5608c3da0 Mon Sep 17 00:00:00 2001 From: bennykok Date: Mon, 9 Dec 2024 16:31:36 +0800 Subject: [PATCH] fix: move update run --- custom_routes.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/custom_routes.py b/custom_routes.py index 5c35897..a3918a8 100644 --- a/custom_routes.py +++ b/custom_routes.py @@ -1251,8 +1251,6 @@ async def send_json_override(self, event, data, sid=None): asyncio.create_task(update_run_ws_event(prompt_id, event, data)) if event == "execution_start": - await update_run(prompt_id, Status.RUNNING) - if prompt_id in prompt_metadata: prompt_metadata[prompt_id].start_time = time.perf_counter() @@ -1265,6 +1263,9 @@ async def send_json_override(self, event, data, sid=None): nodes_start_vram={}, ) + await update_run(prompt_id, Status.RUNNING) + + if event == "executing" and data and CURRENT_START_EXECUTION_DATA: if data.get("node") is None: start_perf_time = CURRENT_START_EXECUTION_DATA.get("start_perf_time")