增加了对minicpm3.0的适配'

2024-09-04 23:10:05 +08:00
parent c5e1736470
commit d7ba97be48
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={