6 lines
405 B
Django/Jinja
6 lines
405 B
Django/Jinja
{# Metadata #}
|
|
{% set stop_strings = ["<|im_start|>", "<|im_end|>"] %}
|
|
|
|
{# Template #}
|
|
{% for message in messages %}{{'<|im_start|>' + message['role'] + '\n' + message['content']}}{% if (loop.last and add_generation_prompt) or not loop.last %}{{ '<|im_end|>' + '\n'}}{% endif %}{% endfor %}
|
|
{% if add_generation_prompt and messages[-1]['role'] != 'assistant' %}{{ '<|im_start|>assistant\n' }}{% endif %}
|