Merge 2a097a28d96cfa452f174d8b433ee7b1bdb2e8a3 into c176b214cc768d41892add4d4f51c5c5627cbf7b

This commit is contained in:
Cyanocitta Yinhao Wang 2025-12-22 13:14:41 -05:00 committed by GitHub
commit 8263f549a6
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