Merge pull request #5372 from LDLINGLINGLING/main

增加了对minicpm3.0的适配' Former-commit-id: 2e3c221d9c87bd59f48648be8878b7b50347280f
2024-09-05 21:35:42 +08:00
parent e7f92d16d8 ce6dcf3600
commit d6ce902d80
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": {