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

Update template.py

Browse Source 
Former-commit-id: 122213a7a7e114b0c390158cac0ae9faeceb2efc
This commit is contained in:
hoshi-hiyouga 2024-05-27 20:51:26 +08:00 committed by GitHub
parent d2c1df7f3d
commit b7b8223230
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,
) )