vocoder weight

Author: prishajain1

src/maxdiffusion/pipelines/ltx2/ltx2_pipeline.py

@@ -543,12 +543,14 @@ def create_model(rngs: nnx.Rngs, config: HyperParameters):
             upsample_initial_channel=1536,
             upsample_rates=(5, 2, 2, 2, 2, 2),
             upsample_kernel_sizes=(11, 4, 4, 4, 4, 4),
+            use_bias_at_final=False,
             rngs=rngs,
             dtype=jnp.float32,
         )
         bwe_generator = Vocoder(
             upsample_initial_channel=512,
             upsample_kernel_sizes=[12, 11, 4, 4, 4],
+            use_bias_at_final=False,
             rngs=rngs,
             dtype=jnp.float32,
         )