Handle tuple output from workflow and ensure tensor extraction in _gpu_logic

core/pipelines/sd_image_pipeline.py

@@ -36,6 +36,9 @@ class SdImagePipeline(BasePipeline):
         initial_objects = {}
         # Execute the workflow; it returns image tensor(s) from the VAE Decode node.
         decoded_images_tensor = WorkflowExecutor.execute_workflow(workflow, initial_objects=initial_objects)
+        # If the node returns a tuple/list, take the first element which holds the tensor.
+        if isinstance(decoded_images_tensor, (list, tuple)):
+            decoded_images_tensor = decoded_images_tensor[0]
 
         # Convert tensors to PIL images, embed metadata and save them to the output directory.
         out_dir = os.path.abspath(OUTPUT_DIR)