fix

Author: prishajain1

src/maxdiffusion/models/ltx2/transformer_ltx2.py

@@ -1179,8 +1179,8 @@ def scan_fn(carry, block_and_mask):
             audio_rotary_emb=audio_rotary_emb,
             ca_video_rotary_emb=video_cross_attn_rotary_emb,
             ca_audio_rotary_emb=audio_cross_attn_rotary_emb,
-            a2v_cross_attention_mask=a2v_cross_attention_mask,
-            v2a_cross_attention_mask=v2a_cross_attention_mask,
+            a2v_cross_attention_mask=encoder_attention_mask,
+            v2a_cross_attention_mask=audio_encoder_attention_mask,
             attention_mask=mask,
             attention_kwargs=attention_kwargs,
             use_cross_timestep=use_cross_timestep,