prince-canuma commited on
Commit
9105fe8
·
verified ·
1 Parent(s): 444a906

Add chat template

Browse files
Files changed (1) hide show
  1. chat_template.jinja +24 -0
chat_template.jinja ADDED
@@ -0,0 +1,24 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {%- set mode = thinking_mode|default('chat') -%}
2
+ <|begin▁of▁sentence|>
3
+ {%- for message in messages -%}
4
+ {%- if message['role'] == 'system' -%}
5
+ {{- message['content'] -}}
6
+ {%- elif message['role'] == 'user' -%}
7
+ <|User|>{{- message['content'] -}}<|Assistant|>
8
+ {%- if loop.last and mode == 'thinking' -%}
9
+ <think>
10
+ {%- else -%}
11
+ </think>
12
+ {%- endif -%}
13
+ {%- elif message['role'] == 'assistant' -%}
14
+ {%- if mode == 'thinking' and message.get('reasoning_content') -%}
15
+ {{- message['reasoning_content'] -}}</think>
16
+ {%- else -%}
17
+ </think>
18
+ {%- endif -%}
19
+ {{- message['content'] -}}<|end▁of▁sentence|>
20
+ {%- endif -%}
21
+ {%- endfor -%}
22
+ {%- if add_generation_prompt and messages[-1]['role'] != 'user' -%}
23
+ <|Assistant|></think>
24
+ {%- endif -%}