vocoder fix

Author: prishajain1

src/maxdiffusion/models/ltx2/vocoder_ltx2.py

@@ -39,7 +39,11 @@ def __init__(
       kernel_size: int = 3,
       stride: int = 1,
       dilations: Sequence[int] = (1, 3, 5),
+      act_fn: str = "leaky_relu",
       leaky_relu_negative_slope: float = 0.1,
+      antialias: bool = False,
+      antialias_ratio: int = 2,
+      antialias_kernel_size: int = 12,
       *,
       rngs: nnx.Rngs,
       dtype: DType = jnp.float32,