perturbed attn = true in transformer

prishajain1 · prishajain1 · commit 9a77820f388e · 2026-04-12T23:23:33.000+05:30
diff --git a/src/maxdiffusion/pipelines/ltx2/ltx2_pipeline.py b/src/maxdiffusion/pipelines/ltx2/ltx2_pipeline.py
@@ -111,6 +111,7 @@ def create_model(rngs: nnx.Rngs, ltx2_config: dict):
     transformer = LTX2VideoTransformer3DModel(
         **ltx2_config,
         spatio_temporal_guidance_blocks=tuple(getattr(config, "spatio_temporal_guidance_blocks", ())),
+        perturbed_attn=getattr(config, "model_name", "") == "ltx2.3",
         rngs=rngs
     )
     return transformer

Original file line number	Diff line number	Diff line change
`@@ -111,6 +111,7 @@ def create_model(rngs: nnx.Rngs, ltx2_config: dict):`
`111`	`111`	`transformer = LTX2VideoTransformer3DModel(`
`112`	`112`	`**ltx2_config,`
`113`	`113`	`spatio_temporal_guidance_blocks=tuple(getattr(config, "spatio_temporal_guidance_blocks", ())),`
	`114`	`+ perturbed_attn=getattr(config, "model_name", "") == "ltx2.3",`
`114`	`115`	`rngs=rngs`
`115`	`116`	`)`
`116`	`117`	`return transformer`