This commit is contained in:
Cyanocitta Yinhao Wang 2025-04-11 09:49:49 -04:00 committed by GitHub
commit 759526a372
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