Commit-Message-Draft-LLM / chat_template.jinja
srikar-v05's picture
Upload model trained with Unsloth
2cd2f26 verified
raw
history blame contribute delete
782 Bytes
{% if 'role' in messages[0] %}{% for message in messages %}{% if message['role'] == 'user' %}{{'<|user|>
' + message['content'] + '<|end|>
'}}{% elif message['role'] == 'assistant' %}{{'<|assistant|>
' + message['content'] + '<|end|>
'}}{% else %}{{'<|' + message['role'] + '|>
' + message['content'] + '<|end|>
'}}{% endif %}{% endfor %}{% if add_generation_prompt %}{{ '<|assistant|>
' }}{% endif %}{% else %}{% for message in messages %}{% if message['from'] == 'human' %}{{'<|user|>
' + message['value'] + '<|end|>
'}}{% elif message['from'] == 'gpt' %}{{'<|assistant|>
' + message['value'] + '<|end|>
'}}{% else %}{{'<|' + message['from'] + '|>
' + message['value'] + '<|end|>
'}}{% endif %}{% endfor %}{% if add_generation_prompt %}{{ '<|assistant|>
' }}{% endif %}{% endif %}