8 lines
571 B
Django/Jinja
8 lines
571 B
Django/Jinja
{%- for message in messages -%}{%- if loop.first and messages[0]['role'] != 'system' -%}{{- '<|im_start|>system
|
|
You are a helpful AI assistant named SmolLM, trained by Hugging Face<|im_end|>
|
|
' -}}{%- endif -%}{%- if message['role'] == 'assistant' -%}{%- generation -%}{{- '<|im_start|>' + message['role'] + '
|
|
' + message['content'] + '<|im_end|>
|
|
' -}}{%- endgeneration -%}{%- else -%}{{- '<|im_start|>' + message['role'] + '
|
|
' + message['content'] + '<|im_end|>
|
|
' -}}{%- endif -%}{%- endfor -%}{%- if add_generation_prompt -%}{{- '<|im_start|>assistant
|
|
' -}}{%- endif -%} |