[Fix] Split if and only if complete eos string shows up (#477)

open-compass · Oct 13, 2023 · 7d9e386 · 7d9e386
1 parent 861942a
commit 7d9e386
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/opencompass/tasks/openicl_eval.py b/opencompass/tasks/openicl_eval.py
@@ -207,7 +207,7 @@ def _extract_role_pred(self, s: str, begin_str: Optional[str],
         if end_str:
             # TODO: Support calling tokenizer for the accurate eos token
             # and avoid such hardcode
-            end_idx = s.find(end_str[:1], start)
+            end_idx = s.find(end_str, start)
             if end_idx != -1:
                 end = end_idx