From 7dd8a7e67ec633a4e8665c6dfa953822e8438651 Mon Sep 17 00:00:00 2001 From: nick Date: Sat, 12 Oct 2024 16:57:37 -0700 Subject: [PATCH] gpu eveent --- custom_routes.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/custom_routes.py b/custom_routes.py index 516e9a5..97b0e96 100644 --- a/custom_routes.py +++ b/custom_routes.py @@ -542,6 +542,7 @@ async def stream_prompt(data, token): prompt_id = data.get("prompt_id") inputs = data.get("inputs") workflow = data.get("workflow") + gpu_event_id = data.get("gpu_event_id", None) # Now it handles directly in here apply_random_seed_to_workflow(workflow_api) @@ -559,6 +560,7 @@ async def stream_prompt(data, token): file_upload_endpoint=data.get("file_upload_endpoint"), workflow_api=workflow_api, token=token, + gpu_event_id=gpu_event_id, ) # log('info', "Begin prompt", prompt=prompt)