fix and ltx2 backward compatibility

prishajain1 · prishajain1 · commit e03eda640eac · 2026-04-09T23:30:19.000+05:30
diff --git a/src/maxdiffusion/pipelines/ltx2/ltx2_pipeline.py b/src/maxdiffusion/pipelines/ltx2/ltx2_pipeline.py
@@ -126,6 +126,7 @@ def create_model(rngs: nnx.Rngs, ltx2_config: dict):
         "num_layers": 48,
         "caption_channels": 4096,
         "audio_caption_channels": 2048,
+        "use_prompt_embeddings": False,
     }
   else:
     ltx2_config = LTX2VideoTransformer3DModel.load_config(config.pretrained_model_name_or_path, subfolder=subfolder)

Original file line number	Diff line number	Diff line change
`@@ -126,6 +126,7 @@ def create_model(rngs: nnx.Rngs, ltx2_config: dict):`
`126`	`126`	`"num_layers": 48,`
`127`	`127`	`"caption_channels": 4096,`
`128`	`128`	`"audio_caption_channels": 2048,`
	`129`	`+ "use_prompt_embeddings": False,`
`129`	`130`	`}`
`130`	`131`	`else:`
`131`	`132`	`ltx2_config = LTX2VideoTransformer3DModel.load_config(config.pretrained_model_name_or_path, subfolder=subfolder)`