Allow 1D masks for 1D latents.

This commit is contained in:
comfyanonymous 2024-11-11 14:44:19 -05:00
parent 2d28b0b479
commit eb476e6ea9
1 changed files with 1 additions and 0 deletions

View File

@ -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: