Add template&support(Not tested)

Former-commit-id: 60bb60c4dc30a9641ddb57a44ef126f0768566c4
2025-12-17 12:20:37 +08:00 · 2024-04-13 04:31:33 +08:00
parent b1ae554c83
commit ba9fc46712
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",
+)