Make ImagePadForOutpaint return a 3 channel mask. (comfyanonymous#8157)

comfyanonymous · web-flow · commit dc46db7aa48a · 2025-05-16T15:15:55.000-04:00
diff --git a/nodes.py b/nodes.py
@@ -1940,7 +1940,7 @@ def expand_image(self, image, left, top, right, bottom, feathering):
 
         mask[top:top + d2, left:left + d3] = t
 
-        return (new_image, mask)
+        return (new_image, mask.unsqueeze(0))
 
 
 NODE_CLASS_MAPPINGS = {
