diff --git a/opencompass/tasks/openicl_eval.py b/opencompass/tasks/openicl_eval.py index d04fb1991..3fa3f67e5 100644 --- 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