Revert copy-paste error fix in audio_to_video_attn heads

Author: Perseus14

src/maxdiffusion/models/ltx2/transformer_ltx2.py

@@ -296,8 +296,8 @@ def __init__(
         rngs=rngs,
         query_dim=dim,
         context_dim=audio_dim,
-        heads=num_attention_heads,
-        dim_head=attention_head_dim,
+        heads=audio_num_attention_heads,
+        dim_head=audio_attention_head_dim,
         dropout=0.0,
         bias=attention_bias,
         out_bias=attention_out_bias,