Merge pull request #5372 from LDLINGLINGLING/main

增加了对minicpm3.0的适配'
2025-12-16 11:50:35 +08:00 · 2024-09-05 21:35:42 +08:00
parent e9bda48c6d 3d3fbaaff9
commit 1274356263
2 changed files with 20 additions and 0 deletions
--- a/src/llamafactory/extras/constants.py
+++ b/src/llamafactory/extras/constants.py
@@ -809,6 +809,17 @@ register_model_group(
 )


+register_model_group(
+    models={
+        "MiniCPM3-4B": {
+            DownloadSource.DEFAULT: "openbmb/MiniCPM3-4B",
+            DownloadSource.MODELSCOPE: "OpenBMB/MiniCPM3-4B",
+        },
+    },
+    template="cpm3",
+)
+
+
 register_model_group(
    models={
        "Mistral-7B-v0.1": {