From 2a097a28d96cfa452f174d8b433ee7b1bdb2e8a3 Mon Sep 17 00:00:00 2001 From: Cyanocitta Yinhao Wang <218019055@link.cuhk.edu.cn> Date: Mon, 10 Mar 2025 15:42:44 +0800 Subject: [PATCH] General way of receiving image for multiple json --- script_examples/websockets_api_example_ws_images.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/script_examples/websockets_api_example_ws_images.py b/script_examples/websockets_api_example_ws_images.py index 6508ecc99..652ce1707 100644 --- a/script_examples/websockets_api_example_ws_images.py +++ b/script_examples/websockets_api_example_ws_images.py @@ -42,7 +42,7 @@ def get_images(ws, prompt): else: current_node = data['node'] else: - if current_node == 'save_image_websocket_node': + if prompt[current_node]['class_type'] == 'SaveImageWebsocket': images_output = output_images.get(current_node, []) images_output.append(out[8:]) output_images[current_node] = images_output