tests: make caching test more comples

huggingface · Jan 10, 2025 · 6a87c8d · 6a87c8d
1 parent f6f12b5
commit 6a87c8d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_environments.py b/tests/test_environments.py
@@ -386,7 +386,7 @@ def test_cached_generate_batched(self):
             [True, True, True], past_key_values, past_attention_masks, past_input_ids
         )
 
-        input_texts2 = [" short interim", " a slightly longer interim", "another interim"]
+        input_texts2 = [" short interim", " a somewhat longer section in between", "something else entirely! So, "]
         model_inputs2 = [self.gpt2_tokenizer(txt, return_tensors="pt").input_ids.squeeze() for txt in input_texts2]
 
         outputs_cached, _, _, _, _ = env._generate_batched(