423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2025-10-15 08:08:09 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update template.py

Browse Source 
Former-commit-id: af869e4c48eb426c4078415533f6dab89123a9d8
This commit is contained in:
hoshi-hiyouga 2024-05-27 20:51:26 +08:00 committed by GitHub
parent 0d9e364a90
commit 1bb1ae834e
1 changed files with 3 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -761,16 +761,15 @@ _register_template(
format_user=StringFormatter( format_user=StringFormatter(
slots=[ slots=[
( (
"<|start_header_id|>GPT4 Correct User<|end_header_id|>\n\n{{content}}<|eot_id|>" "<|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" "<|start_header_id|>GPT4 Correct Assistant<|end_header_id|>\n\n"
) )
] ]
), ),
format_system=StringFormatter( format_system=StringFormatter(slots=[{"bos_token"}, "{{content}}"]),
slots=[{"bos_token"}, "<|start_header_id|>System<|end_header_id|>\n\n{{content}}<|eot_id|>"]
),
stop_words=["<|eot_id|>"], stop_words=["<|eot_id|>"],
replace_eos=True, replace_eos=True,
force_system=True,
) )