fix in lora

Author: prishajain1

src/maxdiffusion/loaders/lora_conversion_utils.py

@@ -758,12 +758,12 @@ def translate_ltx2_nnx_path_to_diffusers_lora(nnx_path_str, scan_layers=False):
 
   # --- 3. Translation Logic ---
   if scan_layers:
-    if nnx_path_str.startswith("blocks."):
-      inner_suffix = nnx_path_str[len("blocks.") :]
+    if nnx_path_str.startswith("transformer_blocks."):
+      inner_suffix = nnx_path_str[len("transformer_blocks.") :]
       if inner_suffix in suffix_map:
         return f"diffusion_model.transformer_blocks.{{}}.{suffix_map[inner_suffix]}"
   else:
-    m = re.match(r"^blocks\.(\d+)\.(.+)$", nnx_path_str)
+    m = re.match(r"^transformer_blocks\.(\d+)\.(.+)$", nnx_path_str)
     if m:
       idx, inner_suffix = m.group(1), m.group(2)
       if inner_suffix in suffix_map: