This commit is contained in:
hiyouga 2024-05-27 20:54:26 +08:00
parent 87ea0a8bcd
commit c1fdf81df6
2 changed files with 2 additions and 2 deletions

View File

@ -742,7 +742,6 @@ _register_template(
_register_template(
name="olmo",
format_user=StringFormatter(slots=["<|user|>\n{{content}}<|assistant|>\n"]),
format_assistant=StringFormatter(slots=["{{content}}", {"eos_token"}]),
format_system=StringFormatter(slots=[{"eos_token"}, "{{content}}"]),
force_system=True,
)
@ -751,11 +750,11 @@ _register_template(
_register_template(
name="openchat",
format_user=StringFormatter(slots=["GPT4 Correct User: {{content}}", {"eos_token"}, "GPT4 Correct Assistant:"]),
format_assistant=StringFormatter(slots=["{{content}}", {"eos_token"}]),
format_system=StringFormatter(slots=[{"bos_token"}, "{{content}}"]),
force_system=True,
)
_register_template(
name="openchat-3.6",
format_user=StringFormatter(

View File

@ -719,6 +719,7 @@ register_model_group(
template="openchat",
)
register_model_group(
models={
"OpenChat3.6-8B-Chat": {