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

gpu eveent

Browse Source
This commit is contained in:
nick 2024-10-12 16:57:37 -07:00
parent 778e6fefe6
commit 7dd8a7e67e
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
custom_routes.py
View File

@ -542,6 +542,7 @@ async def stream_prompt(data, token):
prompt_id = data.get("prompt_id") prompt_id = data.get("prompt_id")
inputs = data.get("inputs") inputs = data.get("inputs")
workflow = data.get("workflow") workflow = data.get("workflow")
gpu_event_id = data.get("gpu_event_id", None)
# Now it handles directly in here # Now it handles directly in here
apply_random_seed_to_workflow(workflow_api) 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"), file_upload_endpoint=data.get("file_upload_endpoint"),
workflow_api=workflow_api, workflow_api=workflow_api,
token=token, token=token,
gpu_event_id=gpu_event_id,
) )
# log('info', "Begin prompt", prompt=prompt) # log('info', "Begin prompt", prompt=prompt)