14 lines
403 B
Django/Jinja
14 lines
403 B
Django/Jinja
{% for m in messages %}
|
|
{% if m['role'] == 'system' %}
|
|
<|im_start|>system
|
|
{{ m['content'] }}<|im_end|>
|
|
{% elif m['role'] == 'user' %}
|
|
<|im_start|>user
|
|
{{ m['content'] }}<|im_end|>
|
|
{% elif m['role'] == 'assistant' %}
|
|
<|im_start|>assistant
|
|
{% generation %}{{ m['content'] }}{{ eos_token }}{% endgeneration %}
|
|
{% endif %}
|
|
{% endfor %}
|
|
{% if add_generation_prompt %}<|im_start|>assistant
|
|
{% endif %} |