24 lines
754 B
Django/Jinja
24 lines
754 B
Django/Jinja
{%- if not add_generation_prompt is defined -%}
|
|
{%- set add_generation_prompt = true -%}
|
|
{%- endif -%}
|
|
{%- if not cls_token is defined -%}
|
|
{%- set cls_token = "<|begin_of_sentence|>" -%}
|
|
{%- endif -%}
|
|
{%- if not sep_token is defined -%}
|
|
{%- set sep_token = "<|end_of_sentence|>" -%}
|
|
{%- endif -%}
|
|
{{- cls_token -}}
|
|
{%- for message in messages -%}
|
|
{%- if message["role"] == "user" -%}
|
|
{{- "User: " + message["content"] + "
|
|
" -}}
|
|
{%- elif message["role"] == "assistant" -%}
|
|
{{- "Assistant: " + message["content"] + sep_token -}}
|
|
{%- elif message["role"] == "system" -%}
|
|
{{- message["content"] + "
|
|
" -}}
|
|
{%- endif -%}
|
|
{%- endfor -%}
|
|
{%- if add_generation_prompt -%}
|
|
{{- "Assistant: " -}}
|
|
{%- endif -%} |