12 lines
500 B
Django/Jinja
12 lines
500 B
Django/Jinja
{{- bos_token }}
|
|
{%- for message in messages %}
|
|
{%- if message['role'] == 'assistant' %}
|
|
{{- '<|start_header_id|>assistant<|end_header_id|>\n\n' }}
|
|
{%- generation %}
|
|
{{- message['content'] | trim + '<|eot_id|>' }}
|
|
{%- endgeneration %}
|
|
{%- else %}
|
|
{{- '<|start_header_id|>' + message['role'] + '<|end_header_id|>\n\n' + message['content'] | trim + '<|eot_id|>' }}
|
|
{%- endif %}
|
|
{%- endfor %}
|
|
{{- '<|start_header_id|>assistant<|end_header_id|>\n\n' }} |