From 268f53dddbda5905859c6facfce90e90736c6f7d Mon Sep 17 00:00:00 2001
From: hoshi-hiyouga <hiyouga@buaa.edu.cn>
Date: Mon, 15 Apr 2024 22:56:55 +0800
Subject: [PATCH] Update constants.py

---
 src/llmtuner/extras/constants.py | 44 ++++++++++++++++----------------
 1 file changed, 22 insertions(+), 22 deletions(-)

diff --git a/src/llmtuner/extras/constants.py b/src/llmtuner/extras/constants.py
index 5b81a9a9..321c36a4 100644
--- a/src/llmtuner/extras/constants.py
+++ b/src/llmtuner/extras/constants.py
@@ -242,6 +242,28 @@ register_model_group(
 )
 
 
+register_model_group(
+    models={
+        "CommandR-35B-Chat": {
+            DownloadSource.DEFAULT: "CohereForAI/c4ai-command-r-v01",
+            DownloadSource.MODELSCOPE: "AI-ModelScope/c4ai-command-r-v01",
+        },
+        "CommandR-Plus-104B-Chat": {
+            DownloadSource.DEFAULT: "CohereForAI/c4ai-command-r-plus",
+            DownloadSource.MODELSCOPE: "AI-ModelScope/c4ai-command-r-plus",
+        },
+        "CommandR-35B-4bit-Chat": {
+            DownloadSource.DEFAULT: "CohereForAI/c4ai-command-r-v01-4bit",
+            DownloadSource.MODELSCOPE: "mirror013/c4ai-command-r-v01-4bit",
+        },
+        "CommandR-Plus-104B-4bit-Chat": {
+            DownloadSource.DEFAULT: "CohereForAI/c4ai-command-r-plus-4bit",
+        },
+    },
+    template="cohere",
+)
+
+
 register_model_group(
     models={
         "DeepSeek-LLM-7B-Base": {
@@ -994,25 +1016,3 @@ 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",
-)
-
-
-register_model_group(
-    models={
-        "C4AI-Command-R-plus-104B": {
-            DownloadSource.DEFAULT: "CohereForAI/c4ai-command-r-plus",
-            DownloadSource.MODELSCOPE: "AI-ModelScope/c4ai-command-r-plus",
-        }
-    },
-    template="c4ai",
-)
\ No newline at end of file