17 lines
395 B
Django/Jinja
17 lines
395 B
Django/Jinja
{%- for message in messages %}
|
|
{%- if message.role == "user" %}
|
|
{{- '<|text_start|>
|
|
' + message.content + '<|text_end|>
|
|
' }}
|
|
{%- elif message.role == "assistant" %}
|
|
{{- '<|corrected_start|>
|
|
' + message.content + '<|corrected_end|>
|
|
' }}
|
|
{%- else %}
|
|
{{- raise('Unknown role: ' + message.role) }}
|
|
{%- endif %}
|
|
{%- endfor %}
|
|
{%- if add_generation_prompt %}
|
|
{{- '<|corrected_start|>
|
|
' }}
|
|
{%- endif %} |