vocoder fix

Author: prishajain1

src/maxdiffusion/models/ltx2/vocoder_ltx2.py

@@ -606,7 +606,13 @@ def __init__(
         upsample_factors=upsample_factors,
         resnet_kernel_sizes=resnet_kernel_sizes,
         resnet_dilations=resnet_dilations,
+        act_fn=act_fn,
         leaky_relu_negative_slope=leaky_relu_negative_slope,
+        antialias=antialias,
+        antialias_ratio=antialias_ratio,
+        antialias_kernel_size=antialias_kernel_size,
+        final_act_fn=final_act_fn,
+        final_bias=final_bias,
         output_sampling_rate=input_sampling_rate,
         rngs=rngs,
         dtype=dtype,
@@ -620,7 +626,13 @@ def __init__(
         upsample_factors=bwe_upsample_factors,
         resnet_kernel_sizes=bwe_resnet_kernel_sizes,
         resnet_dilations=bwe_resnet_dilations,
+        act_fn=bwe_act_fn,
         leaky_relu_negative_slope=bwe_leaky_relu_negative_slope,
+        antialias=bwe_antialias,
+        antialias_ratio=bwe_antialias_ratio,
+        antialias_kernel_size=bwe_antialias_kernel_size,
+        final_act_fn=bwe_final_act_fn,
+        final_bias=bwe_final_bias,
         output_sampling_rate=output_sampling_rate,
         rngs=rngs,
         dtype=dtype,