fix

prishajain1 · prishajain1 · commit c6dc26551edb · 2026-04-11T13:46:25.000+05:30
diff --git a/src/maxdiffusion/models/ltx2/transformer_ltx2.py b/src/maxdiffusion/models/ltx2/transformer_ltx2.py
@@ -1182,7 +1182,6 @@ def scan_fn(carry, block_and_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,
             modality_mask=modality_mask,
         )

Original file line number	Diff line number	Diff line change
`@@ -1182,7 +1182,6 @@ def scan_fn(carry, block_and_mask):`
`1182`	`1182`	`a2v_cross_attention_mask=encoder_attention_mask,`
`1183`	`1183`	`v2a_cross_attention_mask=audio_encoder_attention_mask,`
`1184`	`1184`	`attention_mask=mask,`
`1185`		`- attention_kwargs=attention_kwargs,`
`1186`	`1185`	`use_cross_timestep=use_cross_timestep,`
`1187`	`1186`	`modality_mask=modality_mask,`
`1188`	`1187`	`)`