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

add models

Browse Source
This commit is contained in:
hiyouga
2023-12-18 19:09:31 +08:00
parent 71a9c16171
commit 709ac8870a
3 changed files with 11 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -431,9 +431,13 @@ register_model_group(
register_model_group(
models={
"Phi1.5-1.3B": {
"Phi-1.5-1.3B": {
DownloadSource.DEFAULT: "microsoft/phi-1_5",
DownloadSource.MODELSCOPE: "allspace/PHI_1-5"
},
"Phi-2-2.7B": {
DownloadSource.DEFAULT: "microsoft/phi-2",
DownloadSource.MODELSCOPE: "AI-ModelScope/phi-2"
}
},
module="Wqkv"