add openchat-3.6-8B support

This commit is contained in:
Jianbai Ye committed 2024-05-27 20:42:08 +08:00
1 parent 08bd0440b5
commit cff815391f
2 files changed
+26

No files matched your search

+17
View File
@@ -756,6 +756,23 @@ _register_template(
force_system=True,
)
_register_template(
name="openchat-3.6",
format_user=StringFormatter(
slots=[
(
"<|start_header_id|>GPT4 Correct User<|end_header_id|>\n\n{{content}}<|eot_id|>"
"<|start_header_id|>GPT4 Correct Assistant<|end_header_id|>\n\n"
)
]
),
format_system=StringFormatter(
slots=[{"bos_token"}, "<|start_header_id|>System<|end_header_id|>\n\n{{content}}<|eot_id|>"]
),
stop_words=["<|eot_id|>"],
replace_eos=True,
)
_register_template(
name="orion",