From f6b263a16d12524a4b55b6d9485d716cf402bbd1 Mon Sep 17 00:00:00 2001 From: kijai <40791699+kijai@users.noreply.github.com> Date: Sun, 7 Jan 2024 18:29:39 +0200 Subject: [PATCH] Update nodes.py --- nodes.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/nodes.py b/nodes.py index e824bd5..b28ccf0 100644 --- a/nodes.py +++ b/nodes.py @@ -703,8 +703,7 @@ class GrowMaskWithBlur: # Convert back to tensor out[idx] = pil2tensor(pil_image) - blurred = torch.cat(out, dim=0) - print(blurred.shape) + blurred = torch.stack(out, dim=0) return (blurred, 1.0 - blurred)