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

add modelscope models

Browse Source 
Former-commit-id: 4de3081eea9cede78a1f2db65cf22a5731c54447
This commit is contained in:
hoshi-hiyouga
2024-09-26 11:22:48 +08:00
committed by GitHub
parent b867e164fe
commit 7d4cb79822
1 changed files with 4 additions and 0 deletions
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",