Update nodes.py

2025-12-09 21:04:41 +08:00 · 2024-01-07 18:29:39 +02:00 · 2024-01-07 18:29:39 +02:00 · f6b263a16d
commit f6b263a16d
parent a0a7e8556c
1 changed files with 1 additions and 2 deletions
--- 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)