Set built-in default reasoning system prompt and auto <think> generation
Browse files- chat_template.jinja +22 -1
chat_template.jinja
CHANGED
|
@@ -1 +1,22 @@
|
|
| 1 |
-
{{ bos_token }}{%
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{{ bos_token }}{%- set default_system = 'Реши задачу. Весь ход мыслей и шаги подробно пиши внутри тегов <think>...</think>, а итоговый ответ — после них.' -%}
|
| 2 |
+
{%- if messages[0]['role'] == 'system' -%}
|
| 3 |
+
{%- set system_message = messages[0]['content'] -%}
|
| 4 |
+
{%- set loop_messages = messages[1:] -%}
|
| 5 |
+
{%- else -%}
|
| 6 |
+
{%- set system_message = default_system -%}
|
| 7 |
+
{%- set loop_messages = messages -%}
|
| 8 |
+
{%- endif -%}
|
| 9 |
+
{%- for message in loop_messages -%}
|
| 10 |
+
{%- if message['role'] == 'user' -%}
|
| 11 |
+
{%- if loop.first and system_message -%}
|
| 12 |
+
{{ '[INST] ' + system_message + '\n\n' + message['content'] + ' [/INST]' }}
|
| 13 |
+
{%- else -%}
|
| 14 |
+
{{ '[INST] ' + message['content'] + ' [/INST]' }}
|
| 15 |
+
{%- endif -%}
|
| 16 |
+
{%- elif message['role'] == 'assistant' -%}
|
| 17 |
+
{{ message['content'] + eos_token }}
|
| 18 |
+
{%- endif -%}
|
| 19 |
+
{%- endfor -%}
|
| 20 |
+
{%- if add_generation_prompt -%}
|
| 21 |
+
{{ '<think>\n' }}
|
| 22 |
+
{%- endif -%}
|