ltx2.3 connectors loading

prishajain1 · prishajain1 · commit ab4841a0c389 · 2026-04-09T12:36:42.000+05:30
diff --git a/src/maxdiffusion/pipelines/ltx2/ltx2_pipeline.py b/src/maxdiffusion/pipelines/ltx2/ltx2_pipeline.py
@@ -327,6 +327,8 @@ def create_model(rngs: nnx.Rngs, config: HyperParameters):
         connector_kwargs.update(
             {
                 "video_connector_num_layers": 8,
+                "audio_connector_num_layers": 8,
+                "caption_channels": 2048,
                 "video_gated_attn": True,
                 "per_modality_projections": True,
                 "proj_bias": True,

Original file line number	Diff line number	Diff line change
`@@ -327,6 +327,8 @@ def create_model(rngs: nnx.Rngs, config: HyperParameters):`
`327`	`327`	`connector_kwargs.update(`
`328`	`328`	`{`
`329`	`329`	`"video_connector_num_layers": 8,`
	`330`	`+ "audio_connector_num_layers": 8,`
	`331`	`+ "caption_channels": 2048,`
`330`	`332`	`"video_gated_attn": True,`
`331`	`333`	`"per_modality_projections": True,`
`332`	`334`	`"proj_bias": True,`