fix baichuan template for training #597 #616

Former-commit-id: 0a3f6984259526775b0efdb8a1b0b24f564a7239
This commit is contained in:
hiyouga 2023-08-21 17:41:51 +08:00
parent fa15416024
commit d3bef03dc6

View File

@ -472,9 +472,30 @@ register_template(
r"""
Supports: https://huggingface.co/baichuan-inc/Baichuan-13B-Chat
Used for training.
"""
register_template(
name="baichuan",
prefix=[
"{{system}}"
],
prompt=[
{"token": "<reserved_102>"}, # user token
"{{query}}",
{"token": "<reserved_103>"} # assistant token
],
system="",
sep=[],
stop_words=[]
)
r"""
Supports: https://huggingface.co/baichuan-inc/Baichuan-13B-Chat
Used for inference of the original model.
"""
register_template(
name="baichuan_eval",
prefix=[
"{{system}}",
{"token": "<reserved_102>"} # user token