Skip to content

Commit a60c2c8

Browse files
authored
fix_ziya_template_bug (#303)
1 parent edb7b86 commit a60c2c8

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

swift/llm/utils/template.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -674,7 +674,7 @@ def register_template(template_type: str,
674674
[['eos_token_id']], [['eos_token_id']], ''))
675675
register_template(
676676
TemplateType.ziya,
677-
Template([['bos_token_id', '{{SYSTEM}}']], ['<human>:{{QUERY}}\n<bot>:'],
677+
Template([['bos_token_id'], '{{SYSTEM}}'], ['<human>:{{QUERY}}\n<bot>:'],
678678
['\n'], [['eos_token_id']], ''))
679679

680680
register_template(

0 commit comments

Comments
 (0)