force 4way

prishajain1 · prishajain1 · commit b61e8a4817f1 · 2026-05-05T11:21:25.000+05:30
diff --git a/src/maxdiffusion/pipelines/ltx2/ltx2_pipeline.py b/src/maxdiffusion/pipelines/ltx2/ltx2_pipeline.py
@@ -1844,6 +1844,7 @@ def transformer_forward_pass(
         "scheduler_step",
         "logical_axis_rules",
         "use_cross_timestep",
+        "force_4way",
     ),
 )
 def run_diffusion_loop(

Original file line number	Diff line number	Diff line change
`@@ -1844,6 +1844,7 @@ def transformer_forward_pass(`
`1844`	`1844`	`"scheduler_step",`
`1845`	`1845`	`"logical_axis_rules",`
`1846`	`1846`	`"use_cross_timestep",`
	`1847`	`+ "force_4way",`
`1847`	`1848`	`),`
`1848`	`1849`	`)`
`1849`	`1850`	`def run_diffusion_loop(`