fwizzer1 commited on
Commit
ffab8db
·
verified ·
1 Parent(s): f10f48a

Bake Fwizzer-R1 thinking system prompt into chat_template.jinja

Browse files
Files changed (1) hide show
  1. chat_template.jinja +45 -25
chat_template.jinja CHANGED
@@ -1,25 +1,45 @@
1
- {{ bos_token }}{%- set default_system = 'Ты думающая нейросеть а зовут тебя Fwizzer-R1-3B-RU. Весь ход мыслей и шаги пиши внутри тегов <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 + '
13
-
14
- ' + message['content'] + ' [/INST]' }}
15
- {%- else -%}
16
- {{ '[INST] ' + message['content'] + ' [/INST]' }}
17
- {%- endif -%}
18
- {%- elif message['role'] == 'assistant' -%}
19
- {{ message['content'] + eos_token }}
20
- {%- endif -%}
21
- {%- endfor -%}
22
- {%- if add_generation_prompt -%}
23
- {{ '<think>
24
- ' }}
25
- {%- endif -%}
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {{- bos_token }}
2
+ {%- set default_system_message = 'Ты думающая нейросеть а зовут тебя Fwizzer-R1-3B-RU. Весь ход мыслей и шаги пиши внутри тегов <think>(напиши сначала) и </think>(напиши по окончанию рассуждений), а итоговый ответ — обязательно после них.' %}
3
+ {%- if messages[0]['role'] == 'system' %}
4
+ {{- '[SYSTEM_PROMPT]' -}}
5
+ {%- if messages[0]['content'] is string %}
6
+ {{- messages[0]['content'] -}}
7
+ {%- else %}
8
+ {%- for block in messages[0]['content'] %}
9
+ {%- if block['type'] == 'text' %}
10
+ {{- block['text'] }}
11
+ {%- endif %}
12
+ {%- endfor %}
13
+ {%- endif %}
14
+ {{- '[/SYSTEM_PROMPT]' -}}
15
+ {%- set loop_messages = messages[1:] %}
16
+ {%- else %}
17
+ {{- '[SYSTEM_PROMPT]' + default_system_message + '[/SYSTEM_PROMPT]' }}
18
+ {%- set loop_messages = messages %}
19
+ {%- endif %}
20
+ {%- for message in loop_messages %}
21
+ {%- if message['role'] == 'user' %}
22
+ {%- if message['content'] is string %}
23
+ {{- '[INST]' + message['content'] + '[/INST]' }}
24
+ {%- else %}
25
+ {{- '[INST]' }}
26
+ {%- for block in message['content'] %}
27
+ {%- if block['type'] == 'text' %}
28
+ {{- block['text'] }}
29
+ {%- endif %}
30
+ {%- endfor %}
31
+ {{- '[/INST]' }}
32
+ {%- endif %}
33
+ {%- elif message['role'] == 'assistant' %}
34
+ {%- if message['content'] is string %}
35
+ {{- message['content'] }}
36
+ {%- else %}
37
+ {%- for block in message['content'] %}
38
+ {%- if block['type'] == 'text' %}
39
+ {{- block['text'] }}
40
+ {%- endif %}
41
+ {%- endfor %}
42
+ {%- endif %}
43
+ {{- eos_token }}
44
+ {%- endif %}
45
+ {%- endfor %}