chat_template.jinja

{{ bos_token }}{% for message in messages %}{% if message['role'] == 'system' %}{{ '[INST] System: ' + message['content'] + '\n\n' }}{% elif message['role'] == 'user' %}{% if loop.index0 == 0 or (loop.index0 == 1 and messages[0]['role'] == 'system') %}{{ message['content'] + ' [/INST]' }}{% else %}{{ '[INST] ' + message['content'] + ' [/INST]' }}{% endif %}{% elif message['role'] == 'assistant' %}{{ ' ' + message['content'] + eos_token }}{% else %}{{ raise_exception('Only system, user, and assistant roles are supported!') }}{% endif %}{% endfor %}{% if add_generation_prompt %}{{ ' ' }}{% endif %}