12 lines
435 B
Plaintext
12 lines
435 B
Plaintext
{%- for message in messages %}
|
|
{%- if message['role'] == 'user' %}
|
|
{{- '<|user|>' + message['content'] + eos_token }}
|
|
{%- elif message['role'] == 'system' %}
|
|
{{- '<|system|>' + message['content'] + eos_token }}
|
|
{%- elif message['role'] == 'assistant' %}
|
|
{{- '<|assistant|>' + message['content'] + eos_token }}
|
|
{%- endif %}
|
|
{%- if loop.last and add_generation_prompt %}
|
|
{{- '<|assistant|>' }}
|
|
{%- endif %}
|
|
{%- endfor %} |