mirror of
https://github.com/jags111/efficiency-nodes-comfyui.git
synced 2026-05-06 16:56:44 -03:00
Fix Image Overlay crash on batch image inputs (#365)
* Initial plan * Fix Image Overlay node batch processing by extracting first image Co-authored-by: jags111 <5968619+jags111@users.noreply.github.com> * Add .gitignore and remove pycache files Co-authored-by: jags111 <5968619+jags111@users.noreply.github.com> * Add warning when multiple overlay images are detected Co-authored-by: jags111 <5968619+jags111@users.noreply.github.com> --------- Co-authored-by: copilot-swe-agent[bot] <198982749+Copilot@users.noreply.github.com> Co-authored-by: jags111 <5968619+jags111@users.noreply.github.com>
This commit is contained in:
@@ -3957,6 +3957,12 @@ 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:
|
||||
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)
|
||||
|
||||
|
||||
Reference in New Issue
Block a user