From 768153d0f06ff0a15f3c12dad7064acd2074e755 Mon Sep 17 00:00:00 2001
From: marko1616 <marko1616@outlook.com>
Date: Sat, 13 Apr 2024 07:32:40 +0800
Subject: [PATCH] Add c4ai-command-r-plus link

Former-commit-id: d0705518ee85408364172803ca3dd68e978e829e
---
 src/llmtuner/extras/constants.py | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/src/llmtuner/extras/constants.py b/src/llmtuner/extras/constants.py
index 0cd2f987..cf68a225 100644
--- a/src/llmtuner/extras/constants.py
+++ b/src/llmtuner/extras/constants.py
@@ -1003,4 +1003,14 @@ register_model_group(
         }
     },
     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