From 2b95c2c866399ca1914b4da486fe52808f7a9c60 Mon Sep 17 00:00:00 2001 From: kaidu Date: Sat, 19 Apr 2025 10:47:58 +0200 Subject: [PATCH] fix mask bug --- nodes.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/nodes.py b/nodes.py index feba39e..405f96a 100644 --- a/nodes.py +++ b/nodes.py @@ -250,7 +250,7 @@ class ReduxAdvanced: for t in conditioning: n = [torch.cat((t[0], cond), dim=1), t[1].copy()] c.append(n) - return (c, image, masko and masko.squeeze(-1)) + return (c, image, masko.squeeze(-1) if masko is not None else None) # A dictionary that contains all nodes you want to export with their names