feat(ltx2): pass sharding specs to text projections

Author: Perseus14

src/maxdiffusion/models/ltx2/transformer_ltx2.py

@@ -696,13 +696,15 @@ def __init__(
         hidden_size=inner_dim,
         dtype=self.dtype,
         weights_dtype=self.weights_dtype,
+        sharding_specs=self.sharding_specs,
     )
     self.audio_caption_projection = NNXPixArtAlphaTextProjection(
         rngs=rngs,
         in_features=self.caption_channels,
         hidden_size=audio_inner_dim,
         dtype=self.dtype,
         weights_dtype=self.weights_dtype,
+        sharding_specs=self.sharding_specs,
     )
     # 3. Timestep Modulation Params and Embedding
     self.time_embed = LTX2AdaLayerNormSingle(