mirror of
https://github.com/jags111/efficiency-nodes-comfyui.git
synced 2026-05-06 08:46: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:
37
.gitignore
vendored
Normal file
37
.gitignore
vendored
Normal file
@@ -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
|
||||
@@ -3958,6 +3958,12 @@ class TSC_ImageOverlay:
|
||||
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)
|
||||
|
||||
# Add Alpha channel to overlay
|
||||
|
||||
Reference in New Issue
Block a user