Update README.md

README.md

@@ -39,7 +39,7 @@ text = tokenizer.apply_chat_template(
     messages,
     tokenize=False,
     add_generation_prompt=True,
-    enable_thinking=True # 启用思维模式，可选
+    enable_thinking=True 
 )
 model_inputs = tokenizer([text], return_tensors="pt").to(model.device)
 
@@ -52,7 +52,6 @@ output_ids = generated_ids[0][len(model_inputs.input_ids[0]):].tolist()
 
 # parsing thinking content
 try:
-    # 查找 </think> 标记 (token id: 151668)
     index = len(output_ids) - output_ids[::-1].index(151668)
 except ValueError:
     index = 0
@@ -60,8 +59,8 @@ except ValueError:
 thinking_content = tokenizer.decode(output_ids[:index], skip_special_tokens=True).strip("\n")
 content = tokenizer.decode(output_ids[index:], skip_special_tokens=True).strip("\n")
 
-print("thinking content:", thinking_content)
-print("content:", content)
+print("psychologist thinking content:", thinking_content)
+print("psychologist content:", content)
 ```
 
 ---