testing with interleaved

prishajain1 · prishajain1 · commit 218f16ff810a · 2026-02-11T16:51:00.000+05:30
diff --git a/src/maxdiffusion/tests/ltx2_parity_test.py b/src/maxdiffusion/tests/ltx2_parity_test.py
@@ -415,7 +415,7 @@ def test_import_parity_comparison(self):
           num_layers=1,
           mesh=self.mesh,
           attention_kernel="dot_product",
-          rope_type="split"
+          rope_type="interleaved"
       )
 
     # 2. Convert Weights (PyTorch -> Flax NNX)

Original file line number	Diff line number	Diff line change
`@@ -415,7 +415,7 @@ def test_import_parity_comparison(self):`
`415`	`415`	`num_layers=1,`
`416`	`416`	`mesh=self.mesh,`
`417`	`417`	`attention_kernel="dot_product",`
`418`		`- rope_type="split"`
	`418`	`+ rope_type="interleaved"`
`419`	`419`	`)`
`420`	`420`
`421`	`421`	`# 2. Convert Weights (PyTorch -> Flax NNX)`