diff --git a/src/llmtuner/extras/constants.py b/src/llmtuner/extras/constants.py index 6eca14bf..478a7f99 100644 --- a/src/llmtuner/extras/constants.py +++ b/src/llmtuner/extras/constants.py @@ -410,7 +410,8 @@ register_model_group( DownloadSource.MODELSCOPE: "AI-ModelScope/Mixtral-8x7B-v0.1" }, "Mixtral-8x7B-Chat": { - DownloadSource.DEFAULT: "mistralai/Mixtral-8x7B-Instruct-v0.1" + DownloadSource.DEFAULT: "mistralai/Mixtral-8x7B-Instruct-v0.1", + DownloadSource.MODELSCOPE: "AI-ModelScope/Mixtral-8x7B-Instruct-v0.1" } }, template="mistral" @@ -576,6 +577,10 @@ register_model_group( "XVERSE-13B-Chat": { DownloadSource.DEFAULT: "xverse/XVERSE-13B-Chat", DownloadSource.MODELSCOPE: "xverse/XVERSE-13B-Chat" + }, + "XVERSE-65B-Chat": { + DownloadSource.DEFAULT: "xverse/XVERSE-65B-Chat", + DownloadSource.MODELSCOPE: "xverse/XVERSE-65B-Chat" } }, template="xverse"