fix(ltx2): fix unexpected argument in VAE embeddings call

Author: Perseus14

src/maxdiffusion/models/ltx2/transformer_ltx2.py

@@ -54,10 +54,7 @@ def __init__(
         use_additional_conditions=use_additional_conditions,
         dtype=dtype,
         weights_dtype=weights_dtype,
-        linear_1_kernel_sharding=sharding_specs.net_0_kernel,
-        linear_1_bias_sharding=sharding_specs.net_0_bias,
-        linear_2_kernel_sharding=sharding_specs.net_2_kernel,
-        linear_2_bias_sharding=sharding_specs.net_2_bias,
+        sharding_specs=sharding_specs,
     )
     self.silu = nnx.silu
     self.linear = nnx.Linear(