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