add baichuan template

This commit is contained in:
hiyouga
2023-07-11 18:57:50 +08:00
parent f936a7af0b
commit 1af031c02b
2 changed files with 12 additions and 1 deletions

View File

@@ -125,6 +125,17 @@ class Template:
use_history=True
)
elif self.name == "baichuan":
r"""
Supports: https://huggingface.co/baichuan-inc/Baichuan-13B-Chat
"""
self._register_template(
prefix="",
prompt="<reserved_102>{query}<reserved_103>",
sep="",
use_history=True
)
else:
raise ValueError("Template {} does not exist.".format(self.name))