support Aya23

2024-05-27 20:23:24 +08:00 · 2024-05-27 20:23:24 +08:00 · e626e26446
parent 08bd0440b5
commit e626e26446
1 changed files with 13 additions and 0 deletions
--- a/src/llamafactory/extras/constants.py
+++ b/src/llamafactory/extras/constants.py
@ -86,6 +86,19 @@ def register_model_group(
        VISION_MODELS.add(prefix)


+register_model_group(
+    models={
+        "Aya-23-8B-Chat": {
+            DownloadSource.DEFAULT: "CohereForAI/aya-23-8B",
+        },
+        "Aya-23-35B-Chat": {
+            DownloadSource.DEFAULT: "CohereForAI/aya-23-35B",
+        },
+    },
+    template="cohere",
+)
+
+
 register_model_group(
    models={
        "Baichuan-7B-Base": {