From 44f7e559b907c7d7ef598410315c0aef29da3cbb Mon Sep 17 00:00:00 2001
From: kijai <40791699+kijai@users.noreply.github.com>
Date: Wed, 24 Jul 2024 13:17:47 +0300
Subject: [PATCH] Update image_nodes.py

---
 nodes/image_nodes.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/nodes/image_nodes.py b/nodes/image_nodes.py
index 7a51159..72af5a3 100644
--- a/nodes/image_nodes.py
+++ b/nodes/image_nodes.py
@@ -740,7 +740,7 @@ class MergeImageChannels:
             
             },
             "optional": {
-                "mask": ("MASK", {"default": None}),
+                "alpha": ("MASK", {"default": None}),
                 },
             }
     
@@ -760,7 +760,7 @@ Merges channel data into an image.
         ], dim=-1)
         image = image.squeeze(-2)
         if alpha is not None:
-            image = torch.cat([image, alpha], dim=-1)
+            image = torch.cat([image, alpha.unsqueeze(-1)], dim=-1)
         return (image,)
 
 class ImagePadForOutpaintMasked: