12 lines
321 B
Django/Jinja
12 lines
321 B
Django/Jinja
{{ eos_token }}{% for message in messages %}
|
|
{% if message['role'] == 'user' %}
|
|
{{ '<|user|>
|
|
' + message['content'] }}
|
|
{% elif message['role'] == 'assistant' %}
|
|
{{ '<|assistant|>
|
|
' + message['content'] + eos_token }}
|
|
{% endif %}
|
|
{% if loop.last and add_generation_prompt %}
|
|
{{ '<|assistant|>' }}
|
|
{% endif %}
|
|
{% endfor %} |