[model] support MiniCPM4 (#8314)

2025-06-10 14:38:39 +08:00
parent 31bca4d172
commit b41697c9b6
2 changed files with 59 additions and 0 deletions
--- a/src/llamafactory/extras/constants.py
+++ b/src/llamafactory/extras/constants.py
@@ -1502,6 +1502,25 @@ register_model_group(
    template="cpm3",
 )

+register_model_group(
+    models={
+        "MiniCPM4-0_5B-Chat": {
+            DownloadSource.DEFAULT: "openbmb/MiniCPM4-0.5B",
+            DownloadSource.MODELSCOPE: "OpenBMB/MiniCPM4-0.5B",
+        },
+    },
+    template="cpm4",
+)
+
+register_model_group(
+    models={
+        "MiniCPM4-8B-Chat": {
+            DownloadSource.DEFAULT: "openbmb/MiniCPM4-8B",
+            DownloadSource.MODELSCOPE: "OpenBMB/MiniCPM4-8B",
+        },
+    },
+    template="cpm4",
+)

 register_model_group(
    models={