marcellobullo's picture
End of training
61dccf7 verified
raw
history blame contribute delete
422 Bytes
{% for message in messages %}
{% if message['role'] == 'user' %}
<|user|>
{{ message['content'] }}</s>
{% elif message['role'] == 'assistant' %}
<|assistant|>
{% generation %}
{{ message['content'] }}</s>
{% endgeneration %}
{% endif %}
{% endfor %}
{% if messages[-1]['role'] != 'assistant' %}
<|assistant|>
{% generation %}
{% endgeneration %}
{% endif %}