chat_template.jinja

{%- set begin_transcript = "<custom_token_3><|begin_of_text|>" -%}

{%- set end_transcript = "<|eot_id|><custom_token_4>" -%}

{%- set begin_audio = "<custom_token_5><custom_token_1>" -%}

{%- set end_audio = "<custom_token_2><custom_token_6>" -%}

{%- for message in messages -%}
    {%-   if message.role == 'user' -%}
        {{- begin_transcript + message['content'] | trim + end_transcript -}}
    {%- elif message.role == 'assistant' -%}
        {{- begin_audio + message['content'] | trim + end_audio -}}
    {%- endif -%}
{%- endfor -%}

{%- if add_generation_prompt -%}
    {{- begin_audio -}}
{%- endif -%}