17fbb6a21c
Model: ali-elganzory/SmolLM2-1.7B-SFT-Tulu3-decontaminated-masked Source: Original Platform
17 lines
656 B
Django/Jinja
17 lines
656 B
Django/Jinja
{%- for message in messages -%}
|
|
{%- if message["role"] == "system" -%}
|
|
{{- "<|system|>\n" + message["content"] + "\n" -}}
|
|
{%- elif message["role"] == "user" -%}
|
|
{{- "<|user|>\n" + message["content"] + "\n" -}}
|
|
{%- elif message["role"] == "assistant" -%}
|
|
{%- if not loop.last -%}
|
|
{{- "<|assistant|>\n" -}}{% generation %}{{- message["content"] + eos_token -}}{% endgeneration %}{{- "\n" -}}
|
|
{%- else -%}
|
|
{{- "<|assistant|>\n" -}}{% generation %}{{- message["content"] + eos_token -}}{% endgeneration %}
|
|
{%- endif -%}
|
|
{%- endif -%}
|
|
{%- if loop.last and add_generation_prompt -%}
|
|
{{- "<|assistant|>\n" -}}
|
|
{%- endif -%}
|
|
{%- endfor -%}
|