diff --git a/efficiency_nodes.py b/efficiency_nodes.py index 9e5471e..ff9511c 100644 --- a/efficiency_nodes.py +++ b/efficiency_nodes.py @@ -3960,6 +3960,8 @@ class TSC_ImageOverlay: # Handle batch dimension - use first image if overlay_image is a batch if len(overlay_image.shape) == 4: + if overlay_image.shape[0] > 1: + print(f"{warning('Image Overlay Warning:')} Multiple overlay images detected ({overlay_image.shape[0]}), using only the first image.") overlay_image = overlay_image[0] overlay_image = tensor2pil(overlay_image)