mirror of
https://github.com/oobabooga/text-generation-webui.git
synced 2024-10-01 01:26:03 -04:00
Add an instruction template for Command R
This commit is contained in:
parent
8a8dbf2f16
commit
4e739dc211
26
instruction-templates/Command-R.yaml
Normal file
26
instruction-templates/Command-R.yaml
Normal file
@ -0,0 +1,26 @@
|
||||
instruction_template: |-
|
||||
{%- if messages[0]['role'] == 'system' -%}
|
||||
{%- set loop_messages = messages[1:] -%}
|
||||
{%- set system_message = messages[0]['content'] -%}
|
||||
{%- elif false == true -%}
|
||||
{%- set loop_messages = messages -%}
|
||||
{%- set system_message = 'You are Command-R, a brilliant, sophisticated, AI-assistant trained to assist human users by providing thorough responses. You are trained by Cohere.' -%}
|
||||
{%- else -%}
|
||||
{%- set loop_messages = messages -%}
|
||||
{%- set system_message = false -%}
|
||||
{%- endif -%}
|
||||
{%- if system_message != false -%}
|
||||
{{ '<|START_OF_TURN_TOKEN|><|SYSTEM_TOKEN|>' + system_message + '<|END_OF_TURN_TOKEN|>' }}
|
||||
{%- endif -%}
|
||||
{%- for message in loop_messages -%}
|
||||
{%- set content = message['content'] -%}
|
||||
{%- if message['role'] == 'user' -%}
|
||||
{{ '<|START_OF_TURN_TOKEN|><|USER_TOKEN|>' + content.strip() + '<|END_OF_TURN_TOKEN|>' }}
|
||||
{%- elif message['role'] == 'assistant' -%}
|
||||
{{ '<|START_OF_TURN_TOKEN|><|CHATBOT_TOKEN|>' + content.strip() + '<|END_OF_TURN_TOKEN|>' }}
|
||||
{%- endif -%}
|
||||
{%- endfor -%}
|
||||
{%- if add_generation_prompt -%}
|
||||
{{ '<|START_OF_TURN_TOKEN|><|CHATBOT_TOKEN|>' }}
|
||||
{%- endif -%}
|
||||
|
@ -192,3 +192,5 @@
|
||||
instruction_template: 'Synthia'
|
||||
.*(hercules|hyperion):
|
||||
instruction_template: 'ChatML'
|
||||
.*command-r:
|
||||
instruction_template: 'Command-R'
|
||||
|
Loading…
Reference in New Issue
Block a user