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

Update greenmind_14b_r1_reasoning_parser.py

Browse files
greenmind_14b_r1_reasoning_parser.py CHANGED
@@ -43,8 +43,8 @@ class GreenMind14bR1ReasoningParser(ReasoningParser):
43
 
44
  self.think_start_ids = [13708, 766, 397]
45
  self.think_start_ids_fast = [13708, 766, 29]
46
- self.response_start_ids = [198, 522, 26865, 397, 27, 9217, 29]
47
- self.response_start_ids_fast = [522, 26865, 397, 27, 9217, 29]
48
  self.response_end_ids = [522, 9217, 29]
49
  self.fast_think_ids = [
50
  13708, 766, 1339, 522, 26865, 397, 27, 9217, 397
 
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]
49
  self.fast_think_ids = [
50
  13708, 766, 1339, 522, 26865, 397, 27, 9217, 397