Update tests/generation/test_stopping_criteria.py

Co-authored-by: Joao Gante <[email protected]>
huggingface · Feb 15, 2024 · ee563d8 · ee563d8
1 parent 06941de
commit ee563d8
Showing 1 changed file with 2 additions and 0 deletions.
diff --git 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|",