viethq5 commited on
Commit
a061951
·
verified ·
1 Parent(s): 5fad179

Update greenmind_14b_r1_reasoning_parser.py

Browse files
greenmind_14b_r1_reasoning_parser.py CHANGED
@@ -41,8 +41,8 @@ class GreenMind14bR1ReasoningParser(ReasoningParser):
41
  rf"{self.think_start_expr}(.*?){self.response_start_expr}(.*)",
42
  re.DOTALL)
43
 
44
- self.think_start_ids = [13708, 766, 397]
45
- self.think_start_ids_fast = [13708, 766, 29]
46
  self.response_start_ids = [522, 26865, 397, 27, 9217, 397] # </think>\n<answer>\n
47
  self.response_start_ids_fast = [522, 26865, 397, 27, 9217, 29] # </think>\n<answer>
48
  self.response_end_ids = [522, 9217, 29]
 
41
  rf"{self.think_start_expr}(.*?){self.response_start_expr}(.*)",
42
  re.DOTALL)
43
 
44
+ self.think_start_ids = [13708, 766, 397] # <think>\n
45
+ self.think_start_ids_fast = [27, 26865, 397] # < think >\n
46
  self.response_start_ids = [522, 26865, 397, 27, 9217, 397] # </think>\n<answer>\n
47
  self.response_start_ids_fast = [522, 26865, 397, 27, 9217, 29] # </think>\n<answer>
48
  self.response_end_ids = [522, 9217, 29]