kyy/llm_trainer
1
0 Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

support Yi-34B-Chat models

Browse Source
This commit is contained in:
hiyouga committed 2023-11-23 19:31:49 +08:00
1 parent 5085b00a1d
commit ff1c289229
2 files changed
+22 -2
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

src/llmtuner/extras/constants.py
+6 -2
View File
@@ -264,8 +264,12 @@ register_model_group(
register_model_group(
models={
"Yi-6B": "01-ai/Yi-6B",
"Yi-34B": "01-ai/Yi-34B"
}
"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"
},
template="yi"
)