GrowMask now works with mask batches.

comfy_extras/nodes_mask.py

@@ -327,7 +327,10 @@ class GrowMask:
         kernel = np.array([[c, 1, c],
                            [1, 1, 1],
                            [c, 1, c]])
-        output = mask.numpy().copy()
+        mask = mask.reshape((-1, mask.shape[-2], mask.shape[-1]))
+        out = []
+        for m in mask:
+            output = m.numpy()
             while expand < 0:
                 output = scipy.ndimage.grey_erosion(output, footprint=kernel)
                 expand += 1
@@ -335,7 +338,8 @@ class GrowMask:
                 output = scipy.ndimage.grey_dilation(output, footprint=kernel)
                 expand -= 1
             output = torch.from_numpy(output)
-        return (output,)
+            out.append(output)
+        return (torch.cat(out, dim=0),)