setting gated_attn to true+ weight loading fix

Author: prishajain1

src/maxdiffusion/models/ltx2/ltx2_3_utils.py

@@ -68,6 +68,8 @@ def rename_for_ltx2_3_transformer(key):
   if "audio_text_proj_in" in key:
     key = key.replace("audio_text_proj_in", "feature_extractor.audio_linear")
 
+  key = key.replace("to_gate_logits.weight", "to_gate_logits.kernel")
+
   key = key.replace("k_norm", "norm_k")
   key = key.replace("q_norm", "norm_q")
   key = key.replace("adaln_single", "time_embed")