update

Signed-off-by: root <root@rack-ge107-ya-20250517193228.idczw.hb1.kwaidc.com>

chat_template.jinja

@@ -0,0 +1,88 @@
+{%- if tools %}
+    {{- '<|im_start|>system\n' }}
+    {%- if messages[0]['role'] == 'system' %}
+        {{- messages[0]['content'] }}
+    {%- else %}
+        {{- '' }}
+    {%- endif %}
+    {{- "\n\n# Tools\n\nYou may call one or more functions to assist with the user query.\n\nYou are provided with function signatures within <tools></tools> XML tags:\n<tools>" }}
+    {%- for tool in tools %}
+        {{- "\n" }}
+        {{- tool | tojson }}
+    {%- endfor %}
+    {{- "\n</tools>\n\nFor each function call, return a json object with function name and arguments within <tool_call></tool_call> XML tags:\n<tool_call>\n{\"name\": <function-name>, \"arguments\": <args-json-object>}\n</tool_call><|im_end|>\n" }}
+{%- else %}
+    {%- if messages[0]['role'] == 'system' %}
+        {{- '<|im_start|>system\n' + messages[0]['content'] + '<|im_end|>\n' }}
+    {%- else %}
+        {{- '<|im_start|>system\nYou are a helpful assistant.<|im_end|>\n' }}
+    {%- endif %}
+{%- endif %}
+{%- for message in messages %}
+    {%- if (message.role == "user") or (message.role == "system" and not loop.first) %}
+        {{- '<|im_start|>' + message.role + '\n' + message.content + '<|im_end|>' + '\n' }}
+    {%- elif message.role == "assistant" and not message.tool_calls %}
+        {%- set content = message.content %}
+        {%- if not loop.last %}
+            {%- set answer_blocks = message.content.split('<answer>\n') %}
+            {%- if answer_blocks|length > 1 %}
+                {%- set last_answer_block = answer_blocks[-1] %}
+                {%- if '\n</answer>' in last_answer_block %}
+                    {%- set content = last_answer_block.split('\n</answer>')[0] %}
+                {%- else %}
+                    {%- set content = message.content.split('<think_off>')[-1].lstrip('\n') %}
+                    {%- set content = content.split('</think>')[-1].lstrip('\n') %}
+                {%- endif %}
+            {%- else %}
+                {%- set content = message.content.split('<think_off>')[-1].lstrip('\n') %}
+                {%- set content = content.split('</think>')[-1].lstrip('\n') %}
+            {%- endif %}
+        {%- endif %}
+        {{- '<|im_start|>' + message.role + '\n' + content + '<|im_end|>' + '\n' }}
+    {%- elif message.role == "assistant" %}
+        {%- set content = message.content %}
+        {%- if not loop.last %}
+            {%- set answer_blocks = message.content.split('<answer>\n') %}
+            {%- if answer_blocks|length > 1 %}
+                {%- set last_answer_block = answer_blocks[-1] %}
+                {%- if '\n</answer>' in last_answer_block %}
+                    {%- set content = last_answer_block.split('\n</answer>')[0] %}
+                {%- else %}
+                    {%- set content = message.content.split('<think_off>')[-1].lstrip('\n') %}
+                    {%- set content = content.split('</think>')[-1].lstrip('\n') %}
+                {%- endif %}
+            {%- else %}
+                {%- set content = message.content.split('<think_off>')[-1].lstrip('\n') %}
+                {%- set content = content.split('</think>')[-1].lstrip('\n') %}
+            {%- endif %}
+        {%- endif %}
+        {{- '<|im_start|>' + message.role }}
+        {%- if message.content %}
+            {{- '\n' + content }}
+        {%- endif %}
+        {%- for tool_call in message.tool_calls %}
+            {%- if tool_call.function is defined %}
+                {%- set tool_call = tool_call.function %}
+            {%- endif %}
+            {{- '\n<tool_call>\n{\"name\": \"' }}
+            {{- tool_call.name }}
+            {{- '\", \"arguments\": ' }}
+            {{- tool_call.arguments | tojson }}
+            {{- '}\n</tool_call>' }}
+        {%- endfor %}
+        {{- '<|im_end|>\n' }}
+    {%- elif message.role == "tool" %}
+        {%- if (loop.index0 == 0) or (messages[loop.index0 - 1].role != "tool") %}
+            {{- '<|im_start|>user' }}
+        {%- endif %}
+        {{- '\n<tool_response>\n' }}
+        {{- message.content }}
+        {{- '\n</tool_response>' }}
+        {%- if loop.last or (messages[loop.index0 + 1].role != "tool") %}
+            {{- '<|im_end|>\n' }}
+        {%- endif %}
+    {%- endif %}
+{%- endfor %}
+{%- if add_generation_prompt %}
+    {{- '<|im_start|>assistant\n<judge>\n' }}
+{%- endif %}

search_results_think_in.txt

@@ -1,10 +0,0 @@
-搜索字符串: 'think_in'
-搜索目录: /share/xiangwen/model_new/qwen_1_7b_cpt_autothink/stage_qwen_1_7b_cpt_global_step_1600
-搜索时间: 2025-10-31 05:09:51.341993
-找到 5 个匹配文件:
-
-output/LCBv4v5plus/20251006_193241/predictions/stage_qwen_1_7b_cpt_global_step_1600/lcb_code_generation_0.json
-output/LCBv4v5plus/20251006_193241/predictions/stage_qwen_1_7b_cpt_global_step_1600/lcb_code_generation_1.json
-output/humaneval/20251006_193251/predictions/stage_qwen_1_7b_cpt_global_step_1600/openai_humaneval_1.json
-output/mbpp/20251006_193241/predictions/stage_qwen_1_7b_cpt_global_step_1600/mbpp_passk_0.json
-output/mbpp/20251006_193241/predictions/stage_qwen_1_7b_cpt_global_step_1600/mbpp_passk_1.json

special_tokens_map.json

@@ -1,4 +1,12 @@
 {
+  "additional_special_tokens": [
+    "<judge>",
+    "</judge>",
+    "<answer>",
+    "</answer>",
+    "<think_on>",
+    "<think_off>"
+  ],
   "eos_token": {
     "content": "<|im_end|>",
     "lstrip": false,

tokenizer_config.json

@@ -260,7 +260,6 @@
     }
   },
   "bos_token": null,
-  "chat_template": "{%- if tools %}\n    {{- '<|im_start|>system\\n' }}\n    {%- if messages[0]['role'] == 'system' %}\n        {{- messages[0]['content'] }}\n    {%- else %}\n        {{- '' }}\n    {%- endif %}\n    {{- \"\\n\\n# Tools\\n\\nYou may call one or more functions to assist with the user query.\\n\\nYou are provided with function signatures within <tools></tools> XML tags:\\n<tools>\" }}\n    {%- for tool in tools %}\n        {{- \"\\n\" }}\n        {{- tool | tojson }}\n    {%- endfor %}\n    {{- \"\\n</tools>\\n\\nFor each function call, return a json object with function name and arguments within <tool_call></tool_call> XML tags:\\n<tool_call>\\n{\\\"name\\\": <function-name>, \\\"arguments\\\": <args-json-object>}\\n</tool_call><|im_end|>\\n\" }}\n{%- else %}\n    {%- if messages[0]['role'] == 'system' %}\n        {{- '<|im_start|>system\\n' + messages[0]['content'] + '<|im_end|>\\n' }}\n    {%- else %}\n        {{- '<|im_start|>system\\nYou are a helpful assistant.<|im_end|>\\n' }}\n    {%- endif %}\n{%- endif %}\n{%- for message in messages %}\n    {%- if (message.role == \"user\") or (message.role == \"system\" and not loop.first) %}\n        {{- '<|im_start|>' + message.role + '\\n' + message.content + '<|im_end|>' + '\\n' }}\n    {%- elif message.role == \"assistant\" and not message.tool_calls %}\n        {%- set content = message.content %}\n        {%- if not loop.last %}\n            {%- set answer_blocks = message.content.split('<answer>\\n') %}\n            {%- if answer_blocks|length > 1 %}\n                {%- set last_answer_block = answer_blocks[-1] %}\n                {%- if '\\n</answer>' in last_answer_block %}\n                    {%- set content = last_answer_block.split('\\n</answer>')[0] %}\n                {%- else %}\n                    {%- set content = message.content.split('<think_off>')[-1].lstrip('\\n') %}\n                    {%- set content = content.split('</think>')[-1].lstrip('\\n') %}\n                {%- endif %}\n            {%- else %}\n                {%- set content = message.content.split('<think_off>')[-1].lstrip('\\n') %}\n                {%- set content = content.split('</think>')[-1].lstrip('\\n') %}\n            {%- endif %}\n        {%- endif %}\n        {{- '<|im_start|>' + message.role + '\\n' + content + '<|im_end|>' + '\\n' }}\n    {%- elif message.role == \"assistant\" %}\n        {%- set content = message.content %}\n        {%- if not loop.last %}\n            {%- set answer_blocks = message.content.split('<answer>\\n') %}\n            {%- if answer_blocks|length > 1 %}\n                {%- set last_answer_block = answer_blocks[-1] %}\n                {%- if '\\n</answer>' in last_answer_block %}\n                    {%- set content = last_answer_block.split('\\n</answer>')[0] %}\n                {%- else %}\n                    {%- set content = message.content.split('<think_off>')[-1].lstrip('\\n') %}\n                    {%- set content = content.split('</think>')[-1].lstrip('\\n') %}\n                {%- endif %}\n            {%- else %}\n                {%- set content = message.content.split('<think_off>')[-1].lstrip('\\n') %}\n                {%- set content = content.split('</think>')[-1].lstrip('\\n') %}\n            {%- endif %}\n        {%- endif %}\n        {{- '<|im_start|>' + message.role }}\n        {%- if message.content %}\n            {{- '\\n' + content }}\n        {%- endif %}\n        {%- for tool_call in message.tool_calls %}\n            {%- if tool_call.function is defined %}\n                {%- set tool_call = tool_call.function %}\n            {%- endif %}\n            {{- '\\n<tool_call>\\n{\\\"name\\\": \\\"' }}\n            {{- tool_call.name }}\n            {{- '\\\", \\\"arguments\\\": ' }}\n            {{- tool_call.arguments | tojson }}\n            {{- '}\\n</tool_call>' }}\n        {%- endfor %}\n        {{- '<|im_end|>\\n' }}\n    {%- elif message.role == \"tool\" %}\n        {%- if (loop.index0 == 0) or (messages[loop.index0 - 1].role != \"tool\") %}\n            {{- '<|im_start|>user' }}\n        {%- endif %}\n        {{- '\\n<tool_response>\\n' }}\n        {{- message.content }}\n        {{- '\\n</tool_response>' }}\n        {%- if loop.last or (messages[loop.index0 + 1].role != \"tool\") %}\n            {{- '<|im_end|>\\n' }}\n        {%- endif %}\n    {%- endif %}\n{%- endfor %}\n{%- if add_generation_prompt %}\n    {{- '<|im_start|>assistant\\n<judge>\\n' }}\n{%- endif %}",
   "clean_up_tokenization_spaces": false,
   "eos_token": "<|im_end|>",
   "errors": "replace",