p04798526
/
LLaMA-Factory-310P3
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

tiny fix

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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/llamafactory/data/template.py
View File

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

1
src/llamafactory/extras/constants.py
View File

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