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

tweak

Browse Source
This commit is contained in:
karrix 2024-10-22 13:57:19 +08:00
parent d41c4de352
commit 0a9d0d3e3e
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
custom_routes.py
View File

@ -1217,7 +1217,7 @@ async def send_json_override(self, event, data, sid=None):
await update_run(prompt_id, Status.FAILED)
# await update_run_with_output(prompt_id, data)
if event == "executed" and "node" in data and "std_output" in data:
if event == "executed" and "node" in data and "output" in data:
node_meta = None
if prompt_id in prompt_metadata:
node = data.get("node")
@ -1231,7 +1231,7 @@ async def send_json_override(self, event, data, sid=None):
# else:
await update_run_with_output(
prompt_id,
data.get("std_output"),
data.get("output"),
node_id=data.get("node"),
node_meta=node_meta,
)
@ -1765,7 +1765,10 @@ async def update_run_with_output(
have_upload_media = False
if data is not None:
have_upload_media = (
"images" in data or "files" in data or "gifs" in data or "mesh" in data
"images" in data
and isinstance(data.get("images"), list)
and len(data["images"]) > 0
and data["images"][0].get("type") == "output"
)
if bypass_upload and have_upload_media:
print(