chunzhi/comfyui-deploy
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: move update run

Browse Source
This commit is contained in:
bennykok 2024-12-09 16:31:36 +08:00
parent 0e3baf22df
commit 1d63b21643
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
custom_routes.py
View File

@ -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)) asyncio.create_task(update_run_ws_event(prompt_id, event, data))
if event == "execution_start": if event == "execution_start":
await update_run(prompt_id, Status.RUNNING)
if prompt_id in prompt_metadata: if prompt_id in prompt_metadata:
prompt_metadata[prompt_id].start_time = time.perf_counter() 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={}, nodes_start_vram={},
) )
await update_run(prompt_id, Status.RUNNING)
if event == "executing" and data and CURRENT_START_EXECUTION_DATA: if event == "executing" and data and CURRENT_START_EXECUTION_DATA:
if data.get("node") is None: if data.get("node") is None:
start_perf_time = CURRENT_START_EXECUTION_DATA.get("start_perf_time") start_perf_time = CURRENT_START_EXECUTION_DATA.get("start_perf_time")