5 lines
414 B
Django/Jinja
5 lines
414 B
Django/Jinja
{{- bos_token }}{%- for message in messages %}{{- '<|start_header_id|>' + message['role'] + '<|end_header_id|>
|
|
|
|
' }}{%- if message['role'] == 'assistant' %}{% generation %}{{- message['content'] }}{{- '<|eot_id|>' }}{% endgeneration %}{%- else %}{{- message['content'] }}{{- '<|eot_id|>' }}{%- endif %}{%- endfor %}{%- if add_generation_prompt %}{{- '<|start_header_id|>assistant<|end_header_id|>
|
|
|
|
' }}{%- endif %} |