Fix missing ulysses args in cross-attention blocks

Author: Perseus14

src/maxdiffusion/models/ltx2/transformer_ltx2.py

@@ -257,6 +257,8 @@ def __init__(
         flash_block_sizes=flash_block_sizes,
         sharding_specs=self.sharding_specs,
         gated_attn=gated_attn,
+        ulysses_shards=ulysses_shards,
+        ulysses_attention_chunks=ulysses_attention_chunks,
     )
 
     self.audio_norm2 = nnx.RMSNorm(
@@ -349,6 +351,8 @@ def __init__(
         flash_min_seq_length=flash_min_seq_length,
         sharding_specs=self.sharding_specs,
         gated_attn=gated_attn,
+        ulysses_shards=ulysses_shards,
+        ulysses_attention_chunks=ulysses_attention_chunks,
     )
 
     # 4. Feed Forward