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 solar 10.7B #1907

Browse Source
This commit is contained in:
hiyouga
2024-01-14 00:30:30 +08:00
parent 3c8e72f585
commit bf73224f33
2 changed files with 27 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -534,6 +534,20 @@ register_model_group(
)
register_model_group(
models={
"SOLAR-10.7B": {
DownloadSource.DEFAULT: "upstage/SOLAR-10.7B-v1.0"
},
"SOLAR-10.7B-Chat": {
DownloadSource.DEFAULT: "upstage/SOLAR-10.7B-Instruct-v1.0",
DownloadSource.MODELSCOPE: "AI-ModelScope/SOLAR-10.7B-Instruct-v1.0"
}
},
template="solar"
)
register_model_group(
models={
"Skywork-13B-Base": {