[Generation Test] Make fast test actually fast (huggingface#17661)

elusenji · Jun 12, 2022 · d707ade · d707ade
1 parent f5c9783
commit d707ade
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/generation/test_generation_utils.py b/tests/generation/test_generation_utils.py
@@ -2633,8 +2633,8 @@ def test_constrained_beam_search_example_integration(self):
         self.assertListEqual(outputs, ["Wie alter sind Sie?"])
 
     def test_constrained_beam_search_mixin_type_checks(self):
-        tokenizer = AutoTokenizer.from_pretrained("t5-base")
-        model = AutoModelForSeq2SeqLM.from_pretrained("t5-base")
+        tokenizer = AutoTokenizer.from_pretrained("patrickvonplaten/t5-tiny-random")
+        model = AutoModelForSeq2SeqLM.from_pretrained("patrickvonplaten/t5-tiny-random")
 
         encoder_input_str = "translate English to German: How old are you?"
         input_ids = tokenizer(encoder_input_str, return_tensors="pt").input_ids