fix bug of gguf and support for LiquidAI/LFM2-1.2B (#927)

n1ck-guo · web-flow · commit 80c8de57972f · 2025-10-22T12:54:18.000+08:00
Signed-off-by: n1ck-guo &lt;heng.guo@intel.com&gt;
diff --git a/auto_round/export/export_to_gguf/convert.py b/auto_round/export/export_to_gguf/convert.py
@@ -109,7 +109,7 @@ def get_moe_name(cls, name, new_name):
         "FFN_DOWN_EXP": ["down_proj", "w2", "linear_1"],
         "FFN_UP_EXP": ["up_proj", "w3", "linear_v"],
     }
-    nums = re.findall(r"\d+", name)
+    nums = re.findall(r"\.(\d+)\.", name)
     if len(nums) != 2:
         return name
     name_tmp = name[: -len(".weight")].replace(f".{nums[1]}", "")

Original file line number	Diff line number	Diff line change
`@@ -109,7 +109,7 @@ def get_moe_name(cls, name, new_name):`
`109`	`109`	`"FFN_DOWN_EXP": ["down_proj", "w2", "linear_1"],`
`110`	`110`	`"FFN_UP_EXP": ["up_proj", "w3", "linear_v"],`
`111`	`111`	`}`
`112`		`- nums = re.findall(r"\d+", name)`
	`112`	`+ nums = re.findall(r"\.(\d+)\.", name)`
`113`	`113`	`if len(nums) != 2:`
`114`	`114`	`return name`
`115`	`115`	`name_tmp = name[: -len(".weight")].replace(f".{nums[1]}", "")`