Allow 1D masks for 1D latents.
This commit is contained in:
parent
2d28b0b479
commit
eb476e6ea9
|
@ -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:
|
||||
|
|
Loading…
Reference in New Issue