423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2025-11-06 10:52:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

add model urls

Browse Source 
Former-commit-id: 3139a9fafab246f5461697efd5ed7a6599d85481
This commit is contained in:
hiyouga 2023-12-13 00:09:17 +08:00
parent 071c78d5e2
commit 7c942804f7
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/llmtuner/extras/constants.py
View File

@ -384,7 +384,8 @@ register_model_group(
DownloadSource.MODELSCOPE: "AI-ModelScope/Mistral-7B-Instruct-v0.1" DownloadSource.MODELSCOPE: "AI-ModelScope/Mistral-7B-Instruct-v0.1"
}, },
"Mistral-7B-v0.2-Chat": { "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" template="mistral"
@ -394,7 +395,8 @@ register_model_group(
register_model_group( register_model_group(
models={ models={
"Mixtral-8x7B": { "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": { "Mixtral-8x7B-Chat": {
DownloadSource.DEFAULT: "mistralai/Mixtral-8x7B-Instruct-v0.1" DownloadSource.DEFAULT: "mistralai/Mixtral-8x7B-Instruct-v0.1"