lucas
/
LLaMA-Factory
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

add modelscope models

This commit is contained in:
hoshi-hiyouga 2024-09-26 11:22:48 +08:00 committed by GitHub
parent 885a0b77ab
commit 8e5d12c2c4
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/llamafactory/extras/constants.py
View File

@ -802,15 +802,19 @@ register_model_group(
models={ models={
"LLaMA3.2-1B": { "LLaMA3.2-1B": {
DownloadSource.DEFAULT: "meta-llama/Llama-3.2-1B", DownloadSource.DEFAULT: "meta-llama/Llama-3.2-1B",
DownloadSource.MODELSCOPE: "LLM-Research/Llama-3.2-1B",
}, },
"LLaMA3.2-3B": { "LLaMA3.2-3B": {
DownloadSource.DEFAULT: "meta-llama/Llama-3.2-3B", DownloadSource.DEFAULT: "meta-llama/Llama-3.2-3B",
DownloadSource.MODELSCOPE: "LLM-Research/Llama-3.2-3B",
}, },
"LLaMA3.2-1B-Instruct": { "LLaMA3.2-1B-Instruct": {
DownloadSource.DEFAULT: "meta-llama/Llama-3.2-1B-Instruct", DownloadSource.DEFAULT: "meta-llama/Llama-3.2-1B-Instruct",
DownloadSource.MODELSCOPE: "LLM-Research/Llama-3.2-1B-Instruct",
}, },
"LLaMA3.2-3B-Instruct": { "LLaMA3.2-3B-Instruct": {
DownloadSource.DEFAULT: "meta-llama/Llama-3.2-3B-Instruct", DownloadSource.DEFAULT: "meta-llama/Llama-3.2-3B-Instruct",
DownloadSource.MODELSCOPE: "LLM-Research/Llama-3.2-3B-Instruct",
}, },
}, },
template="llama3", template="llama3",