Update greenmind_14b_r1_reasoning_parser.py

greenmind_14b_r1_reasoning_parser.py

@@ -30,8 +30,8 @@ class GreenMind14bR1ReasoningParser(ReasoningParser):
         self.think_start_expr = r"<think>\n"
         self.think_end_expr = r"\n</think>\n"
 
-        self.response_start_expr = r"\n</think>\n<answer>\n"
-        self.response_end_expr = r"\n</answer>"
+        self.response_start_expr = r"\n</think>\n<answer>"
+        self.response_end_expr = r"</answer>"
 
         self.full_match_reasoning_regex = re.compile(
             rf"(?:{self.think_start_expr}(.*?){self.response_start_expr})?(.*?){self.response_end_expr}",
@@ -43,9 +43,9 @@ class GreenMind14bR1ReasoningParser(ReasoningParser):
 
         self.think_start_ids = [13708, 766, 397]
         self.think_start_ids_fast = [13708, 766, 29]
-        self.response_start_ids = [198, 522, 26865, 397, 27, 9217, 397]
-        self.response_start_ids_fast = [522, 26865, 397, 27, 9217, 397]
-        self.response_end_ids = [198, 522, 9217, 29]
+        self.response_start_ids = [198, 522, 26865, 397, 27, 9217, 29]
+        self.response_start_ids_fast = [522, 26865, 397, 27, 9217, 29]
+        self.response_end_ids = [522, 9217, 29]
         self.fast_think_ids = [
             13708, 766, 1339, 522, 26865, 397, 27, 9217, 397
         ]