update UI, fix #212

This commit is contained in:
hiyouga
2023-07-20 22:09:06 +08:00
parent c3fcb67486
commit 4d1641c1bf
12 changed files with 155 additions and 45 deletions

View File

@@ -215,7 +215,7 @@ Supports: https://huggingface.co/baichuan-inc/Baichuan-13B-Chat
register_template(
name="baichuan",
prefix="",
prompt=" <reserved_102> {query} <reserved_103> ",
prompt="<reserved_102>{query}<reserved_103>",
sep="</s>",
use_history=True
)