Update nodes.py

This commit is contained in:
kijai 2024-03-22 21:25:10 +02:00
parent 7cc091c1d0
commit f6a65e32d3

View File

@ -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