Skip to content

Commit 459c82e

Browse files
committed
a2v and v2a attn mask to none
1 parent f2e664a commit 459c82e

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

src/maxdiffusion/models/ltx2/transformer_ltx2.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1193,8 +1193,8 @@ def scan_fn(carry, block_mask_and_id):
11931193
audio_rotary_emb=audio_rotary_emb,
11941194
ca_video_rotary_emb=video_cross_attn_rotary_emb,
11951195
ca_audio_rotary_emb=audio_cross_attn_rotary_emb,
1196-
a2v_cross_attention_mask=encoder_attention_mask,
1197-
v2a_cross_attention_mask=audio_encoder_attention_mask,
1196+
a2v_cross_attention_mask=None,
1197+
v2a_cross_attention_mask=None,
11981198
perturbation_mask=mask,
11991199
modality_mask=modality_mask,
12001200
)

0 commit comments

Comments
 (0)