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

13e875cc · Patrick von Platen · GitHub · b4eef63a · 13e875cc
Unverified Commit 13e875cc authored Jun 10, 2022 by Patrick von Platen Committed by GitHub Jun 10, 2022
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

tests/generation/test_generation_utils.py tests/generation/test_generation_utils.py +2 -2

No files found.
--- a/tests/generation/test_generation_utils.py
+++ b/tests/generation/test_generation_utils.py
@@ -2633,8 +2633,8 @@ class GenerationIntegrationTests(unittest.TestCase):
        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