vocoder bwe changes

Author: prishajain1

src/maxdiffusion/models/ltx2/vocoder_bwe_ltx2.py

@@ -222,7 +222,7 @@ def __call__(self, x: Array) -> Array:
         lhs=x_transposed,
         rhs=w_flipped,
         window_strides=(1,),
-        padding="VALID",
+        padding=((0, 0),),
         lhs_dilation=(self.ratio,),
         feature_group_count=self.channels,
         dimension_numbers=jax.lax.ConvDimensionNumbers(