support Yi-34B-Chat models

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

No files matched your search

+16
View File
@@ -682,6 +682,22 @@ register_template(
)
register_template(
name="yi",
prefix=[
"{{system}}"
],
prompt=[
"<|im_start|>user\n{{query}}<|im_end|>\n<|im_start|>assistant\n"
],
system="",
sep=[
"<|im_end|>\n"
],
efficient_eos=True
)
register_template(
name="zephyr",
prefix=[