From c23cc63d3d3c4fd8edd6c3b3ca1a2a32ec328d7d Mon Sep 17 00:00:00 2001
From: hiyouga <467089858@qq.com>
Date: Thu, 6 Jun 2024 03:42:50 +0800
Subject: [PATCH] add codestral 22B

---
 src/llamafactory/extras/constants.py | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/src/llamafactory/extras/constants.py b/src/llamafactory/extras/constants.py
index 4d7685c5..687e16cc 100644
--- a/src/llamafactory/extras/constants.py
+++ b/src/llamafactory/extras/constants.py
@@ -301,6 +301,16 @@ register_model_group(
 )
 
 
+register_model_group(
+    models={
+        "Codestral-22B-v0.1-Chat": {
+            DownloadSource.DEFAULT: "mistralai/Codestral-22B-v0.1",
+        },
+    },
+    template="mistral",
+)
+
+
 register_model_group(
     models={
         "CommandR-35B-Chat": {