423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2025-12-14 19:06:26 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix baichuan template for training #597 #616

Browse Source
This commit is contained in:
hiyouga
2023-08-21 17:41:51 +08:00
parent 5c052836a0
commit 0a3f698425
1 changed files with 21 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
src/llmtuner/extras/template.py
View File

@@ -472,9 +472,30 @@ register_template(
r""" r"""
Supports: https://huggingface.co/baichuan-inc/Baichuan-13B-Chat Supports: https://huggingface.co/baichuan-inc/Baichuan-13B-Chat
Used for training.
""" """
register_template( register_template(
name="baichuan", 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=[ prefix=[
"{{system}}", "{{system}}",
{"token": "<reserved_102>"} # user token {"token": "<reserved_102>"} # user token