garrison commited on
Commit
91df4e8
·
verified ·
1 Parent(s): 6ea4a44

Upload chat_template.jinja with huggingface_hub

Browse files
Files changed (1) hide show
  1. 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 %}