Fix Image Overlay node batch processing by extracting first image

Co-authored-by: jags111 <5968619+jags111@users.noreply.github.com>
2026-02-03 17:03:43 +00:00
parent ed4f801c51
commit 08c19f47fe
3 changed files with 4 additions and 0 deletions
@@ -3957,6 +3957,10 @@ class TSC_ImageOverlay:
            samples = overlay_image.movedim(-1, 1)
            overlay_image = comfy.utils.common_upscale(samples, overlay_image_size[0], overlay_image_size[1], resize_method, False)
            overlay_image = overlay_image.movedim(1, -1)
+        
+        # Handle batch dimension - use first image if overlay_image is a batch
+        if len(overlay_image.shape) == 4:
+            overlay_image = overlay_image[0]
            
        overlay_image = tensor2pil(overlay_image)