Allow 1D masks for 1D latents.

Comfy-Org · Nov 11, 2024 · eb476e6 · eb476e6
1 parent 2d28b0b
commit eb476e6
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/comfy/utils.py b/comfy/utils.py
@@ -853,6 +853,7 @@ def reshape_mask(input_mask, output_shape):
     dims = len(output_shape) - 2
 
     if dims == 1:
+        mask = input_mask
         scale_mode = "linear"
 
     if dims == 2: