Warecube commited on
Commit
362c32c
·
verified ·
1 Parent(s): aaf9908

upload chat_template.jinja

Browse files
Files changed (1) hide show
  1. chat_template.jinja +24 -0
chat_template.jinja ADDED
@@ -0,0 +1,24 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {%- if not messages %}
2
+ {{- raise_exception('No messages provided.') }}
3
+ {%- endif %}
4
+ {%- for message in messages %}
5
+ {%- if loop.first and message.role == 'system' %}
6
+ {{- '<|im_start|>system\n' + message.content | trim + '<|im_end|>\n' }}
7
+ {%- elif message.role == 'user' %}
8
+ {{- '<|im_start|>user\n' + message.content | trim + '<|im_end|>\n' }}
9
+ {%- elif message.role == 'assistant' %}
10
+ {%- if not loop.last %}
11
+ {{- '<|im_start|>assistant\n' + message.content | trim + '<|im_end|>\n' }}
12
+ {%- else %}
13
+ {%- set content = message.content | trim %}
14
+ {%- if content.startswith('<think>') %}
15
+ {{- '<|im_start|>assistant\n' + content + '<|im_end|>\n' }}
16
+ {%- else %}
17
+ {{- '<|im_start|>assistant\n<think>\n' + content + '<|im_end|>\n' }}
18
+ {%- endif %}
19
+ {%- endif %}
20
+ {%- endif %}
21
+ {%- endfor %}
22
+ {%- if add_generation_prompt %}
23
+ {{- '<|im_start|>assistant\n<think>\n' }}
24
+ {%- endif %}