feat: add constants for 9g

2024-10-22 17:25:10 +08:00 · 2024-10-22 17:25:10 +08:00 · 3c3bc98f5e
parent 9ba88b0295
commit 3c3bc98f5e
1 changed files with 9 additions and 0 deletions
--- a/src/llamafactory/extras/constants.py
+++ b/src/llamafactory/extras/constants.py
@ -804,6 +804,15 @@ register_model_group(
    template="cpm",
 )

+register_model_group(
+    models={
+        "CPM-9G-8B": {
+            DownloadSource.DEFAULT: "NotImplemented/UseFilePath",
+            DownloadSource.MODELSCOPE: "NotImplemented/UseFilePath",
+        },
+    },
+    template="cpm",
+)

 register_model_group(
    models={