15 lines
410 B
Plaintext
15 lines
410 B
Plaintext
|
{% for message in messages %}
|
||
|
|
{% if message['role'] == 'user' %}
|
||
|
|
{{ '<|user|>
|
||
|
|
' + message['content'] + eos_token }}
|
||
|
|
{% elif message['role'] == 'system' %}
|
||
|
|
{{ '<|system|>
|
||
|
|
' + message['content'] + eos_token }}
|
||
|
|
{% elif message['role'] == 'assistant' %}
|
||
|
|
{{ '<|assistant|>
|
||
|
|
' + message['content'] + eos_token }}
|
||
|
|
{% endif %}
|
||
|
|
{% if loop.last and add_generation_prompt %}
|
||
|
|
{{ '<|assistant|>' }}
|
||
|
|
{% endif %}
|
||
|
|
{% endfor %}
|