diff --git a/tests/generation/test_stopping_criteria.py b/tests/generation/test_stopping_criteria.py index 4c6878882bb7e7..ec19512bf279f1 100644 --- a/tests/generation/test_stopping_criteria.py +++ b/tests/generation/test_stopping_criteria.py @@ -122,6 +122,8 @@ def test_stop_string_criteria(self): "<|im_start|><|im_end|<|im_end|", "<|im_end|><|im_start|>", "<|im_end|<>stop<|im_end|", + "en d", + "eNd", ] too_short_strings = [ "<|im_end|",