--- a +++ b/chat_templates/mistral-instruct.jinja @@ -0,0 +1,19 @@ +{% if messages[0]['role'] == 'system' %} + {% set system_message = messages[0]['content'] | trim + '\n\n' %} + {% set messages = messages[1:] %} +{% else %} + {% set system_message = '' %} +{% endif %} + +{{ bos_token + system_message}} +{% for message in messages %} + {% if (message['role'] == 'user') != (loop.index0 % 2 == 0) %} + {{ raise_exception('Conversation roles must alternate user/assistant/user/assistant/...') }} + {% endif %} + + {% if message['role'] == 'user' %} + {{ '[INST] ' + message['content'] | trim + ' [/INST]' }} + {% elif message['role'] == 'assistant' %} + {{ ' ' + message['content'] | trim + eos_token }} + {% endif %} +{% endfor %} \ No newline at end of file