modified-chat-gemstone / chat_template.jinja
WillHeld's picture
Upload tokenizer
e9b8209 verified
raw
history blame contribute delete
611 Bytes
{{ bos_token }}
{%- for message in messages -%}
{%- if message['role'] == 'system' -%}
<|system|>
{{ message['content'] | trim }}
{%- elif message['role'] == 'user' -%}
<|user|>
{{ message['content'] | trim }}
{%- elif message['role'] == 'assistant' -%}
{%- if not loop.last -%}
<|assistant|>
{% generation %}{{ message['content'] | trim }}{{ eos_token }}{% endgeneration %}
{%- else -%}
<|assistant|>
{% generation %}{{ message['content'] | trim }}{{ eos_token }}{% endgeneration %}
{%- endif -%}
{%- endif -%}
{%- endfor -%}
{%- if add_generation_prompt -%}
<|assistant|>
{%- endif -%}