args val changed

prishajain1 · prishajain1 · commit abc1575ead46 · 2026-02-09T21:09:37.000+05:30
diff --git a/src/maxdiffusion/tests/ltx_2_transformer_test.py b/src/maxdiffusion/tests/ltx_2_transformer_test.py
@@ -271,6 +271,7 @@ def test_transformer_3d_model_dot_product_attention(self):
             num_attention_heads=8,
             attention_head_dim=128,
             cross_attention_dim=32,
+            caption_channels=32,
             audio_in_channels=self.audio_in_channels,
             audio_out_channels=self.audio_in_channels,
             audio_patch_size=1,
