From 45d37879c233c8b56867bc935deb794808909c41 Mon Sep 17 00:00:00 2001 From: bennykok Date: Fri, 23 Feb 2024 15:09:21 -0800 Subject: [PATCH] fix: not returning images in websocket output node --- comfy-nodes/output_websocket_image.py | 7 +------ 1 file changed, 1 insertion(+), 6 deletions(-) diff --git a/comfy-nodes/output_websocket_image.py b/comfy-nodes/output_websocket_image.py index 6277b0a..751118f 100644 --- a/comfy-nodes/output_websocket_image.py +++ b/comfy-nodes/output_websocket_image.py @@ -37,7 +37,6 @@ class ComfyDeployWebscoketImageOutput: CATEGORY = "output" def run(self, output_id, images, client_id): - results = [] prompt_server = PromptServer.instance loop = prompt_server.loop @@ -51,12 +50,8 @@ class ComfyDeployWebscoketImageOutput: schedule_coroutine_blocking(send_image, ["PNG", image, None], client_id) 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": {}}