instruction_template: |- | |
{%- set ns = namespace(found=false) -%} | |
{%- for message in messages -%} | |
{%- if message['role'] == 'system' -%} | |
{%- set ns.found = true -%} | |
{%- endif -%} | |
{%- endfor -%} | |
{%- if not ns.found -%} | |
{{-'SYSTEM: ' + '' + '\n' -}} | |
{%- endif %} | |
{%- for message in messages %} | |
{%- if message['role'] == 'system' -%} | |
{{-'SYSTEM: ' + message['content'] + '\n' -}} | |
{%- else -%} | |
{%- if message['role'] == 'user' -%} | |
{{-'USER: ' + message['content'] + '\n'-}} | |
{%- else -%} | |
{{-'ASSISTANT: ' + message['content'] + '</s>\n' -}} | |
{%- endif -%} | |
{%- endif -%} | |
{%- endfor -%} | |
{%- if add_generation_prompt -%} | |
{{-'ASSISTANT:'-}} | |
{%- endif -%} | |