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

lopgs

Browse Source
This commit is contained in:
nick 2024-10-07 14:16:55 -07:00
parent 8882f4983c
commit 66d1e42409
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
custom_routes.py
View File

@ -468,6 +468,7 @@ async def comfy_deploy_run(request):
if len(parts) == 2 and parts[0].lower() == "bearer": if len(parts) == 2 and parts[0].lower() == "bearer":
token = parts[1] token = parts[1]
print("RECIEVED DATA", data)
# In older version, we use workflow_api, but this has inputs already swapped in nextjs frontend, which is tricky # In older version, we use workflow_api, but this has inputs already swapped in nextjs frontend, which is tricky
workflow_api = data.get("workflow_api_raw") workflow_api = data.get("workflow_api_raw")
# The prompt id generated from comfy deploy, can be None # The prompt id generated from comfy deploy, can be None
@ -1291,6 +1292,11 @@ async def update_run_ws_event(prompt_id: str, event: str, data: dict):
token = prompt_metadata[prompt_id].token token = prompt_metadata[prompt_id].token
gpu_event_id = prompt_metadata[prompt_id].gpu_event_id or None gpu_event_id = prompt_metadata[prompt_id].gpu_event_id or None
print("prompt_metadata", prompt_metadata[prompt_id])
print("gpu_event_id", gpu_event_id)
print("event", event)
print("data", data)
body = { body = {
"run_id": prompt_id, "run_id": prompt_id,
"ws_event": { "ws_event": {