checkpoint-2000-split2/chat_template.jinja

{%- set default_system_prompt = 'You are a helpful assistant.' %}
    {%- if messages[0]['role'] != 'system' %}
    {{- '<|im_start|>system
' + default_system_prompt + '<|im_end|>
' }}
    {%- endif %}
    {%- for message in messages %}
    {%- if message['role'] == 'system' %}
    {{- '<|im_start|>system
' + message['content'] + '<|im_end|>
' }}
    {%- elif message['role'] == 'user' %}
    {{- '<|im_start|>user
' + message['content'] + '<|im_end|>
' }}
    {%- elif message['role'] == 'assistant' %}
    {{- '<|im_start|>assistant
' }}{% generation %}{{ message['content'] }}{% endgeneration %}{{- '<|im_end|>' }}
    {%- endif %}
    {%- endfor %}
    {%- if add_generation_prompt %}
    {{- '<|im_start|>assistant
' }}
    {%- endif %}