audio_path_size

Author: prishajain1

src/maxdiffusion/tests/ltx_2_transformer_test.py

@@ -273,8 +273,8 @@ def test_transformer_3d_model_dot_product_attention(self):
             cross_attention_dim=32,
             audio_in_channels=self.audio_in_channels,
             audio_out_channels=self.audio_in_channels,
-            audio_patch_size=self.audio_patch_size,
-            audio_patch_size_t=self.audio_patch_size_t,
+            audio_patch_size=1,
+            audio_patch_size_t=1,
             audio_num_attention_heads=8,
             audio_attention_head_dim=128,
             audio_cross_attention_dim=32,