Merge pull request #26 from huixingyun/fix/mask-output

bugfix: reshape mask to (B, H, W)

nodes.py

@@ -249,7 +249,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)
+        return (c, image, masko.squeeze(-1))
 
 
 # A dictionary that contains all nodes you want to export with their names
@@ -263,4 +263,4 @@ NODE_CLASS_MAPPINGS = {
 NODE_DISPLAY_NAME_MAPPINGS = {
     "StyleModelApplySimple": "Apply style model (simple)",
     "ReduxAdvanced": "Apply Redux model (advanced)"
-}
+}