4 lines
577 B
Django/Jinja
4 lines
577 B
Django/Jinja
{{ bos_token }}{% if messages[0]['role'] == 'system' %}{% set messages = messages[1:] %}{% endif %}{% for message in messages %}{% if (message['role'] == 'user') != (loop.index0 % 2 == 0) %}{{ raise_exception('Conversation roles must alternate user/assistant/user/assistant/...') }}{% endif %}{% if (message['role'] == 'assistant') %}{% set role = 'model' %}{% else %}{% set role = message['role'] %}{% endif %}{{ '<start_of_turn>' + role + '
|
|
' + message['content'] | trim + '<end_of_turn>
|
|
' }}{% endfor %}{% if add_generation_prompt %}{{'<start_of_turn>model
|
|
'}}{% endif %} |