Skip to content

Commit 9878b72

Browse files
author
xusenlin
committed
Support for baichuan2 system message
1 parent 8f9f878 commit 9878b72

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

api/adapter/template.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -755,6 +755,11 @@ def template(self) -> str:
755755
<reserved_102>{Prompt}<reserved_103>{Answer}<reserved_102>{Prompt}<reserved_103>
756756
"""
757757
return (
758+
"{% if messages[0]['role'] == 'system' %}"
759+
"{{ messages[0]['content'] }}"
760+
"{% else %}"
761+
"{{ system_prompt }}"
762+
"{% endif %}"
758763
"{% for message in messages %}"
759764
"{% if message['role'] == 'user' %}"
760765
"{{ '<reserved_102>' + message['content'] + '<reserved_103>' }}"

0 commit comments

Comments
 (0)