Upload chat_template.jinja with huggingface_hub
Browse files- chat_template.jinja +15 -0
chat_template.jinja
ADDED
|
@@ -0,0 +1,15 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{% if 'role' in messages[0] %}{% for message in messages %}{% if message['role'] == 'user' %}{{'<|im_start|>user
|
| 2 |
+
' + message['content'] + '<|im_end|>
|
| 3 |
+
'}}{% elif message['role'] == 'assistant' %}{{'<|im_start|>assistant
|
| 4 |
+
' + message['content'] + '<|im_end|>
|
| 5 |
+
' }}{% else %}{{ '<|im_start|>system
|
| 6 |
+
' + message['content'] + '<|im_end|>
|
| 7 |
+
' }}{% endif %}{% endfor %}{% if add_generation_prompt %}{{ '<|im_start|>assistant
|
| 8 |
+
' }}{% endif %}{% else %}{% for message in messages %}{% if message['from'] == 'human' %}{{'<|im_start|>user
|
| 9 |
+
' + message['value'] + '<|im_end|>
|
| 10 |
+
'}}{% elif message['from'] == 'gpt' %}{{'<|im_start|>assistant
|
| 11 |
+
' + message['value'] + '<|im_end|>
|
| 12 |
+
' }}{% else %}{{ '<|im_start|>system
|
| 13 |
+
' + message['value'] + '<|im_end|>
|
| 14 |
+
' }}{% endif %}{% endfor %}{% if add_generation_prompt %}{{ '<|im_start|>assistant
|
| 15 |
+
' }}{% endif %}{% endif %}
|