dealignai commited on
Commit
001b418
·
verified ·
1 Parent(s): 281abfd

chat_template: add enable_thinking flag (sync with JANG_2L)

Browse files
Files changed (1) hide show
  1. chat_template.jinja +5 -1
chat_template.jinja CHANGED
@@ -155,5 +155,9 @@
155
 
156
  {#- Generation prompt -#}
157
  {%- if add_generation_prompt -%}
158
- {{- ']~b]ai' ~ '\n' ~ '<think>' ~ '\n' }}
 
 
 
 
159
  {%- endif -%}
 
155
 
156
  {#- Generation prompt -#}
157
  {%- if add_generation_prompt -%}
158
+ {{- ']~b]ai' ~ '\n' }}
159
+ {%- if enable_thinking is defined and enable_thinking is false -%}
160
+ {%- else -%}
161
+ {{- '<think>' ~ '\n' }}
162
+ {%- endif -%}
163
  {%- endif -%}