mirror of
https://github.com/SakanaAI/doc-to-lora.git
synced 2026-05-30 14:35:15 +02:00
24 lines
1,014 B
Django/Jinja
24 lines
1,014 B
Django/Jinja
{{- bos_token }}
|
|
{%- if messages[0]['role'] == 'system' %}
|
|
{%- set system_message = messages[0]['content'] %}
|
|
{%- set loop_messages = messages[1:] %}
|
|
{%- else %}
|
|
{%- set loop_messages = messages %}
|
|
{%- endif %}
|
|
{% for message in loop_messages %}
|
|
{% if (message['role'] == 'assistant') %}
|
|
{% set role = 'model' %}
|
|
{% else %}
|
|
{% set role = message['role'] %}
|
|
{% endif %}
|
|
{%- if message['role'] == 'user' and loop.first and system_message is defined %}
|
|
{{ '<start_of_turn>' + role + '\n' + system_message + '\n\n' + message['content'] | trim + '<end_of_turn>\n' }}
|
|
{%- elif message['role'] == 'assistant' %}
|
|
{{ '<start_of_turn>' + role + '\n' }}{% generation %}{{ message['content'] + '<end_of_turn>' }}{% endgeneration %}{{ '\n' }}
|
|
{%- else %}
|
|
{{ '<start_of_turn>' + role + '\n' + message['content'] | trim + '<end_of_turn>\n' }}
|
|
{%- endif %}
|
|
{% endfor %}
|
|
{% if add_generation_prompt %}
|
|
{{'<start_of_turn>model\n'}}
|
|
{% endif %}
|