This commit is contained in:
Cyanocitta Yinhao Wang 2026-05-24 16:00:25 -07:00 committed by GitHub
commit 2a7fa4d88f
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -42,7 +42,7 @@ def get_images(ws, prompt):
else: else:
current_node = data['node'] current_node = data['node']
else: else:
if current_node == 'save_image_websocket_node': if prompt[current_node]['class_type'] == 'SaveImageWebsocket':
images_output = output_images.get(current_node, []) images_output = output_images.get(current_node, [])
images_output.append(out[8:]) images_output.append(out[8:])
output_images[current_node] = images_output output_images[current_node] = images_output