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
|
dims = len(output_shape) - 2
|
||||||
|
|
||||||
if dims == 1:
|
if dims == 1:
|
||||||
|
mask = input_mask
|
||||||
scale_mode = "linear"
|
scale_mode = "linear"
|
||||||
|
|
||||||
if dims == 2:
|
if dims == 2:
|
||||||
|
|
Loading…
Reference in New Issue