5 lines
516 B
Django/Jinja
5 lines
516 B
Django/Jinja
{% if messages[0]['role'] == 'system' %}{% set system_message = messages[0]['content'] %}{% set messages = messages[1:] %}{% else %}{% set system_message = '' %}{% endif %}{% for message in messages %}{% if message['role'] == 'user' %}{% if loop.first and system_message %}[INST] <<SYS>>
|
|
{{ system_message }}
|
|
<</SYS>>
|
|
|
|
{{ message['content'] }} [/INST] {% else %}[INST] {{ message['content'] }} [/INST] {% endif %}{% elif message['role'] == 'assistant' %}{{ message['content'] }}{{ eos_token }}{% endif %}{% endfor %} |