vocoder weight

prishajain1 · prishajain1 · commit e1560e08d16c · 2026-04-09T14:08:53.000+05:30
diff --git a/src/maxdiffusion/pipelines/ltx2/ltx2_pipeline.py b/src/maxdiffusion/pipelines/ltx2/ltx2_pipeline.py
@@ -541,6 +541,8 @@ def create_model(rngs: nnx.Rngs, config: HyperParameters):
         # Manually construct for LTX-2.3 to support BWE and avoid TypeError
         base_vocoder = Vocoder(
             upsample_initial_channel=1536,
+            upsample_rates=(5, 2, 2, 2, 2, 2),
+            upsample_kernel_sizes=(11, 4, 4, 4, 4, 4),
             rngs=rngs,
             dtype=jnp.float32,
         )

Original file line number	Diff line number	Diff line change
`@@ -541,6 +541,8 @@ def create_model(rngs: nnx.Rngs, config: HyperParameters):`
`541`	`541`	`# Manually construct for LTX-2.3 to support BWE and avoid TypeError`
`542`	`542`	`base_vocoder = Vocoder(`
`543`	`543`	`upsample_initial_channel=1536,`
	`544`	`+ upsample_rates=(5, 2, 2, 2, 2, 2),`
	`545`	`+ upsample_kernel_sizes=(11, 4, 4, 4, 4, 4),`
`544`	`546`	`rngs=rngs,`
`545`	`547`	`dtype=jnp.float32,`
`546`	`548`	`)`