fix(ltx2): fix unexpected arguments in self.ff call

Author: Perseus14

src/maxdiffusion/models/ltx2/transformer_ltx2.py

@@ -315,10 +315,7 @@ def __init__(
         activation_fn=activation_fn,
         dtype=dtype,
         weights_dtype=weights_dtype,
-        net_0_kernel_sharding=self.sharding_specs.net_0_kernel,
-        net_0_bias_sharding=self.sharding_specs.net_0_bias,
-        net_2_kernel_sharding=self.sharding_specs.net_2_kernel,
-        net_2_bias_sharding=self.sharding_specs.net_2_bias,
+        sharding_specs=self.sharding_specs,
     )
 
     self.audio_norm3 = nnx.RMSNorm(