From f6a65e32d3d80a793eb2defddea3ee49ab314b52 Mon Sep 17 00:00:00 2001
From: kijai <40791699+kijai@users.noreply.github.com>
Date: Fri, 22 Mar 2024 21:25:10 +0200
Subject: [PATCH] Update nodes.py

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

diff --git a/nodes.py b/nodes.py
index 017b89e..63214e4 100644
--- a/nodes.py
+++ b/nodes.py
@@ -757,7 +757,7 @@ class GrowMaskWithBlur:
         kernel = np.array([[c, 1, c],
                            [1, 1, 1],
                            [c, 1, c]])
-        growmask = mask.reshape((-1, mask.shape[-2], mask.shape[-1]))
+        growmask = mask.reshape((-1, mask.shape[-2], mask.shape[-1])).cpu()
         out = []
         previous_output = None
         current_expand = expand