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

fix baichuan template for training #597 #616

Browse Source 
Former-commit-id: 6530c1d972301eac9ef058b3235618bb09833f15
This commit is contained in:
hiyouga 2023-08-21 17:41:51 +08:00
parent 8eec1d50e1
commit 516df9ecce
1 changed files with 21 additions and 0 deletions
Show Stats 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