Allow 1D masks for 1D latents.

This commit is contained in:
comfyanonymous 2024-11-11 14:44:19 -05:00
parent 474fe8bc28
commit 24432ca567

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: