21 lines
613 B
Django/Jinja
21 lines
613 B
Django/Jinja
{%- if messages[0]['role'] != 'system' -%}
|
|
{%- set messages = [{'role': 'system', 'content': 'Du bist ein hilfreicher KI-Assistent.'}] + messages -%}
|
|
{%- endif -%}
|
|
{%- 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' -%}
|
|
{% generation %}{{ '<|assistant|>
|
|
' + message['content'] + eos_token }}{% endgeneration %}
|
|
{%- endif -%}
|
|
{%- endfor -%}
|
|
{%- if add_generation_prompt -%}
|
|
{{ '<|assistant|>
|
|
' }}
|
|
{%- endif -%}
|
|
|