Fix non-hashable static argument in run_diffusion_loop

prishajain1 · prishajain1 · commit de4566506f6d · 2026-04-17T23:09:01.000+05:30
diff --git a/src/maxdiffusion/pipelines/ltx2/ltx2_pipeline.py b/src/maxdiffusion/pipelines/ltx2/ltx2_pipeline.py
@@ -1365,7 +1365,7 @@ def __call__(
           batch_size,
           self.transformer.scan_layers,
           self.scheduler.step,
-          tuple(self.config.logical_axis_rules),
+          tuple(tuple(rule) if isinstance(rule, list) else rule for rule in self.config.logical_axis_rules),
       )
 
     # 8. Decode Latents

Original file line number	Diff line number	Diff line change
`@@ -1365,7 +1365,7 @@ def __call__(`
`1365`	`1365`	`batch_size,`
`1366`	`1366`	`self.transformer.scan_layers,`
`1367`	`1367`	`self.scheduler.step,`
`1368`		`- tuple(self.config.logical_axis_rules),`
	`1368`	`+ tuple(tuple(rule) if isinstance(rule, list) else rule for rule in self.config.logical_axis_rules),`
`1369`	`1369`	`)`
`1370`	`1370`
`1371`	`1371`	`# 8. Decode Latents`