From 4579b7d6076b2870998a08f5d37883fbc8261ff2 Mon Sep 17 00:00:00 2001
From: Copilot <198982749+Copilot@users.noreply.github.com>
Date: Tue, 31 Mar 2026 19:46:59 -0400
Subject: [PATCH] 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>
---
 .gitignore          | 37 +++++++++++++++++++++++++++++++++++++
 efficiency_nodes.py |  6 ++++++
 2 files changed, 43 insertions(+)
 create mode 100644 .gitignore

diff --git a/.gitignore b/.gitignore
new file mode 100644
index 0000000..c6f9226
--- /dev/null
+++ b/.gitignore
@@ -0,0 +1,37 @@
+# Python
+__pycache__/
+*.py[cod]
+*$py.class
+*.so
+.Python
+build/
+develop-eggs/
+dist/
+downloads/
+eggs/
+.eggs/
+lib/
+lib64/
+parts/
+sdist/
+var/
+wheels/
+*.egg-info/
+.installed.cfg
+*.egg
+
+# Virtual environments
+venv/
+ENV/
+env/
+
+# IDE
+.vscode/
+.idea/
+*.swp
+*.swo
+*~
+
+# OS
+.DS_Store
+Thumbs.db
diff --git a/efficiency_nodes.py b/efficiency_nodes.py
index 286d84d..ff9511c 100644
--- a/efficiency_nodes.py
+++ b/efficiency_nodes.py
@@ -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)