From f9ab303629789e0f0163e054a00df5732b682c81 Mon Sep 17 00:00:00 2001
From: hiyouga <hiyouga@buaa.edu.cn>
Date: Wed, 13 Dec 2023 00:09:17 +0800
Subject: [PATCH] add model urls

Former-commit-id: 3552035d7eecff86943f02aa26693544fe295f49
---
 src/llmtuner/extras/constants.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/src/llmtuner/extras/constants.py b/src/llmtuner/extras/constants.py
index f73df95d..e6cb7c38 100644
--- a/src/llmtuner/extras/constants.py
+++ b/src/llmtuner/extras/constants.py
@@ -384,7 +384,8 @@ register_model_group(
             DownloadSource.MODELSCOPE: "AI-ModelScope/Mistral-7B-Instruct-v0.1"
         },
         "Mistral-7B-v0.2-Chat": {
-            DownloadSource.DEFAULT: "mistralai/Mistral-7B-Instruct-v0.2"
+            DownloadSource.DEFAULT: "mistralai/Mistral-7B-Instruct-v0.2",
+            DownloadSource.MODELSCOPE: "AI-ModelScope/Mistral-7B-Instruct-v0.2"
         }
     },
     template="mistral"
@@ -394,7 +395,8 @@ register_model_group(
 register_model_group(
     models={
         "Mixtral-8x7B": {
-            DownloadSource.DEFAULT: "mistralai/Mixtral-8x7B-v0.1"
+            DownloadSource.DEFAULT: "mistralai/Mixtral-8x7B-v0.1",
+            DownloadSource.MODELSCOPE: "AI-ModelScope/Mixtral-8x7B-v0.1"
         },
         "Mixtral-8x7B-Chat": {
             DownloadSource.DEFAULT: "mistralai/Mixtral-8x7B-Instruct-v0.1"