fix

Author: prishajain1

src/maxdiffusion/tests/ltx_2_transformer_test.py

@@ -241,10 +241,8 @@ def test_transformer_3d_model_instantiation_and_forward(self):
                 num_frames=self.num_frames,
                 height=self.height,
                 width=self.width,
-                width=self.width,
                 audio_num_frames=128,
                 fps=24.0,
-                fps=24.0,
                 return_dict=True,
                 encoder_attention_mask=encoder_attention_mask,
                 audio_encoder_attention_mask=audio_encoder_attention_mask
@@ -315,13 +313,10 @@ def test_scan_remat_parity(self):
             audio_encoder_hidden_states=audio_encoder_hidden_states,
             timestep=timestep,
             num_frames=self.num_frames,
-            height=self.height,
             width=self.width,
             height=self.height,
-            width=self.width,
             audio_num_frames=128,
             fps=24.0,
-            fps=24.0,
             return_dict=True,
         )