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

add model urls

Browse Source 
Former-commit-id: 3552035d7e
This commit is contained in:
hiyouga
2023-12-13 00:09:17 +08:00
parent fdaba1648a
commit f9ab303629
1 changed files with 4 additions and 2 deletions
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"