17 lines
656 B
Plaintext
17 lines
656 B
Plaintext
|
{%- for message in messages -%}
|
||
|
|
{%- if message["role"] == "system" -%}
|
||
|
|
{{- "<|system|>\n" + message["content"] + "\n" -}}
|
||
|
|
{%- elif message["role"] == "user" -%}
|
||
|
|
{{- "<|user|>\n" + message["content"] + "\n" -}}
|
||
|
|
{%- elif message["role"] == "assistant" -%}
|
||
|
|
{%- if not loop.last -%}
|
||
|
|
{{- "<|assistant|>\n" -}}{% generation %}{{- message["content"] + eos_token -}}{% endgeneration %}{{- "\n" -}}
|
||
|
|
{%- else -%}
|
||
|
|
{{- "<|assistant|>\n" -}}{% generation %}{{- message["content"] + eos_token -}}{% endgeneration %}
|
||
|
|
{%- endif -%}
|
||
|
|
{%- endif -%}
|
||
|
|
{%- if loop.last and add_generation_prompt -%}
|
||
|
|
{{- "<|assistant|>\n" -}}
|
||
|
|
{%- endif -%}
|
||
|
|
{%- endfor -%}
|