Add template&support(Not tested)

Former-commit-id: 60bb60c4dc30a9641ddb57a44ef126f0768566c4
2024-04-13 04:31:33 +08:00
parent 106a0104da
commit f328413646
2 changed files with 18 additions and 0 deletions
--- a/src/llmtuner/extras/constants.py
+++ b/src/llmtuner/extras/constants.py
@@ -994,3 +994,13 @@ register_model_group(
    },
    template="atom",
 )
+
+register_model_group(
+    models={
+        "C4AI-Command-R-35B": {
+            DownloadSource.DEFAULT: "CohereForAI/c4ai-command-r-v01",
+            DownloadSource.MODELSCOPE: "AI-ModelScope/c4ai-command-r-v01",
+        }
+    },
+    template="c4ai",
+)