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