neutts-2e / chat_template.jinja
suryatmodulus's picture
Duplicate from neuphonic/neutts-2e
b3f916a
Raw
History Blame Contribute Delete
379 Bytes
{%- for message in messages -%}
{%- if message["role"] == "system" -%}
{{- message["content"] -}}
{%- elif message["role"] == "user" -%}
{{- "<|TEXT_PROMPT_START|>" -}}
{{- message["content"] -}}
{{- "<|TEXT_PROMPT_END|>" -}}
{%- elif message["role"] == "assistant" -%}
{{- "<|SPEECH_GENERATION_START|>" -}}
{{- message["content"] -}}
{%- endif -%}
{%- endfor -%}