using ltx2adalayernormsingle

Author: prishajain1

src/maxdiffusion/models/ltx2/transformer_ltx2.py

@@ -257,7 +257,7 @@ def __init__(
         rope_type=rope_type,
         flash_block_sizes=flash_block_sizes,
         flash_min_seq_length=0,
-        gated_attn=cross_attn_mod,
+        gated_attn=False,
     )
 
     self.video_to_audio_norm = nnx.RMSNorm(
@@ -285,7 +285,7 @@ def __init__(
         rope_type=rope_type,
         flash_block_sizes=flash_block_sizes,
         flash_min_seq_length=flash_min_seq_length,
-        gated_attn=cross_attn_mod,
+        gated_attn=False,
     )
 
     # 4. Feed Forward