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

fix #1659

Browse Source 
Former-commit-id: e4123129aae59f4123d53c1f5320e3d5e09ae26d
This commit is contained in:
hiyouga
2023-11-28 20:52:28 +08:00
parent 569860d7ac
commit 670ee3934f
4 changed files with 12 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -217,7 +217,11 @@ register_model_group(
"Qwen-7B": "Qwen/Qwen-7B",
"Qwen-14B": "Qwen/Qwen-14B",
"Qwen-7B-Chat": "Qwen/Qwen-7B-Chat",
"Qwen-14B-Chat": "Qwen/Qwen-14B-Chat"
"Qwen-14B-Chat": "Qwen/Qwen-14B-Chat",
"Qwen-7B-int8-Chat": "Qwen/Qwen-7B-Chat-Int8",
"Qwen-7B-int4-Chat": "Qwen/Qwen-7B-Chat-Int4",
"Qwen-14B-int8-Chat": "Qwen/Qwen-14B-Chat-Int8",
"Qwen-14B-int4-Chat": "Qwen/Qwen-14B-Chat-Int4"
},
module="c_attn",
template="qwen"
@@ -266,8 +270,7 @@ register_model_group(
"Yi-6B": "01-ai/Yi-6B",
"Yi-34B": "01-ai/Yi-34B",
"Yi-34B-Chat": "01-ai/Yi-34B-Chat",
"Yi-34B-int8-Chat": "01-ai/Yi-34B-Chat-8bits",
"Yi-34B-int4-Chat": "01-ai/Yi-34B-Chat-4bits"
"Yi-34B-int8-Chat": "01-ai/Yi-34B-Chat-8bits"
},
template="yi"
)