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 = [
|
| 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]
|