test

Author: Perseus14

src/maxdiffusion/models/lora_nnx.py

@@ -122,7 +122,11 @@ def parse_lora_dict(state_dict, dtype):
       continue
 
     # Standard LoRA
-    m = re.match(r"^(.*?)\.(lora_down|lora_up)\.weight$", k)
+    m = re.match(r"^(.*?)_lora\.(down|up)\.weight$", k)
+    if not m:
+      m = re.match(r"^(.*?)\.lora\.(down|up)\.weight$", k)
+    if not m:
+      m = re.match(r"^(.*?)\.(lora_down|lora_up)\.weight$", k)
     if not m:
       m = re.match(r"^(.*?)\.(lora_A|lora_B)\.weight$", k)