some fix in ltx2_utils.py

Author: prishajain1

src/maxdiffusion/models/ltx2/ltx2_utils.py

@@ -91,7 +91,7 @@ def replace_suffix(lst, old, new):
           lst[-1] = new
       return lst
 
-  if "transformer_blocks" in flax_key_str:
+  if "transformer_blocks" in flax_key_str or ("attn1" in flax_key_str and len(flax_key_str) >= 2 and "attn1" in flax_key_str[-2]):
       if flax_key_str[-1] == "query":
           flax_key_str[-1] = "to_q"
       elif flax_key_str[-1] == "key":