From dc46db7aa48acd035dfc10730a064a9c994fb76b Mon Sep 17 00:00:00 2001
From: comfyanonymous <121283862+comfyanonymous@users.noreply.github.com>
Date: Fri, 16 May 2025 12:15:55 -0700
Subject: [PATCH] Make ImagePadForOutpaint return a 3 channel mask. (#8157)

---
 nodes.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/nodes.py b/nodes.py
index 0a9db1393..95e831b8b 100644
--- a/nodes.py
+++ b/nodes.py
@@ -1940,7 +1940,7 @@ class ImagePadForOutpaint:
 
         mask[top:top + d2, left:left + d3] = t
 
-        return (new_image, mask)
+        return (new_image, mask.unsqueeze(0))
 
 
 NODE_CLASS_MAPPINGS = {