423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2025-12-16 11:50:35 +08:00
Code Issues Packages Projects Releases Wiki Activity

support llama3

Browse Source
This commit is contained in:
hoshi-hiyouga
2024-04-19 01:13:50 +08:00
parent 942362d008
commit 2aaaede247
4 changed files with 45 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -513,6 +513,25 @@ register_model_group(
)
register_model_group(
models={
"LLaMA3-8B": {
DownloadSource.DEFAULT: "meta-llama/Meta-Llama-3-8B",
},
"LLaMA3-70B": {
DownloadSource.DEFAULT: "meta-llama/Meta-Llama-3-70B",
},
"LLaMA3-8B-Chat": {
DownloadSource.DEFAULT: "meta-llama/Meta-Llama-3-8B-Instruct",
},
"LLaMA3-70B-Chat": {
DownloadSource.DEFAULT: "meta-llama/Meta-Llama-3-70B-Instruct",
},
},
template="llama3",
)
register_model_group(
models={
"Mistral-7B-v0.1": {