fix cohere system

This commit is contained in:
hiyouga 2024-05-29 20:58:23 +08:00
parent 0930f58699
commit d0aa36b8ad
1 changed files with 7 additions and 2 deletions

View File

@ -544,8 +544,13 @@ _register_template(
) )
] ]
), ),
format_system=StringFormatter(slots=[{"bos_token"}, "{{content}}"]), format_system=StringFormatter(
force_system=True, slots=[{"bos_token"}, "<|START_OF_TURN_TOKEN|><|SYSTEM_TOKEN|>{{content}}<|END_OF_TURN_TOKEN|>"]
),
default_system=(
"You are Command-R, a brilliant, sophisticated, AI-assistant trained to assist human users "
"by providing thorough responses. You are trained by Cohere."
),
) )