Chat template

#1
by EloyOn - opened

{%- for message in messages -%}
{%- if message["role"] == "system" -%}
{{- "<|im_start|>system<|im_sep|>" + message["content"] + "<|im_end|>" -}}
{%- elif message["role"] == "user" -%}
{{- "<|im_start|>user<|im_sep|>" + message["content"] + "<|im_end|>" -}}
{%- elif message["role"] == "assistant" -%}
{{- "<|im_start|>assistant<|im_sep|>" + message["content"] + "<|im_end|>" -}}
{%- endif -%}
{%- endfor -%}
{%- if add_generation_prompt -%}
{{- "<|im_start|>assistant<|im_sep|>" -}}
{%- endif -%}

EloyOn pinned discussion

Sign up or log in to comment