mirror of
https://github.com/SakanaAI/doc-to-lora.git
synced 2026-05-30 14:35:15 +02:00
18 lines
763 B
Text
18 lines
763 B
Text
|
{%- if messages[0].role == 'system' %}
|
||
|
|
{{- '<|im_start|>system\n' + messages[0].content + '<|im_end|>\n' }}
|
||
|
|
{%- endif %}
|
||
|
|
{%- for message in messages %}
|
||
|
|
{%- if message.content is string %}
|
||
|
|
{%- set content = message.content %}
|
||
|
|
{%- else %}
|
||
|
|
{%- set content = '' %}
|
||
|
|
{%- endif %}
|
||
|
|
{%- if (message.role == "user") or (message.role == "system" and not loop.first) %}
|
||
|
|
{{- '<|im_start|>' + message.role + '\n' + content + '<|im_end|>' + '\n' }}
|
||
|
|
{%- elif message.role == "assistant" %}
|
||
|
|
{{- '<|im_start|>' + message.role + '\n' }}{% generation %}{{ content + '<|im_end|>' }}{% endgeneration %}{{ '\n' }}
|
||
|
|
{%- endif %}
|
||
|
|
{%- endfor %}
|
||
|
|
{%- if add_generation_prompt %}
|
||
|
|
{{- '<|im_start|>assistant\n' }}
|
||
|
|
{%- endif %}
|