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

fix: not returning images in websocket output node

Browse Source
This commit is contained in:
bennykok 2024-02-23 15:09:21 -08:00
parent ddbf6848a7
commit 45d37879c2
1 changed files with 1 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
comfy-nodes/output_websocket_image.py
View File

@ -37,7 +37,6 @@ class ComfyDeployWebscoketImageOutput:
CATEGORY = "output" CATEGORY = "output"
def run(self, output_id, images, client_id): def run(self, output_id, images, client_id):
results = []
prompt_server = PromptServer.instance prompt_server = PromptServer.instance
loop = prompt_server.loop loop = prompt_server.loop
@ -51,12 +50,8 @@ class ComfyDeployWebscoketImageOutput:
schedule_coroutine_blocking(send_image, ["PNG", image, None], client_id) schedule_coroutine_blocking(send_image, ["PNG", image, None], client_id)
print("Image sent") print("Image sent")
# loop.run_until_complete(send_image(["PNG", image, None], client_id))
results.append(
{"source": "websocket", "content-type": "image/png", "type": "output"}
)
return {"ui": {"images": results}} return {"ui": {}}