增加了对minicpm3.0的适配'

Former-commit-id: 4ad3a761af2452ef3f6c61190b7e47c9ea5227b9
2024-09-04 23:10:05 +08:00
parent 2c1eef34cb
commit c692339020
2 changed files with 16 additions and 1 deletions
--- a/src/llamafactory/extras/constants.py
+++ b/src/llamafactory/extras/constants.py
@@ -806,6 +806,15 @@ register_model_group(
    template="cpm",
 )

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

 register_model_group(
    models={