Skip to content

Commit 044e72f

Browse files
remove cond_mask
1 parent f112eab commit 044e72f

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

src/diffusers/pipelines/cosmos/pipeline_cosmos2_5_predict.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -462,8 +462,9 @@ def prepare_latents(
462462

463463
num_cond_latent_frames = (num_frames_in - 1) // self.vae_scale_factor_temporal + 1
464464
cond_indicator = latents.new_zeros(1, 1, latents.size(2), 1, 1)
465-
cond_indicator[:, :, 0:num_cond_latent_frames] = 1.0
466-
cond_mask = cond_indicator * ones_padding + (1 - cond_indicator) * zeros_padding
465+
# cond_indicator[:, :, 0:num_cond_latent_frames] = 1.0
466+
# cond_mask = cond_indicator * ones_padding + (1 - cond_indicator) * zeros_padding
467+
cond_mask = zeros_padding
467468

468469
return (
469470
latents,

0 commit comments

Comments
 (0)