From 8bfe7c8d708a0d24f2fc0d854e5bead0baf43119 Mon Sep 17 00:00:00 2001 From: Joao Gante Date: Fri, 20 Sep 2024 16:47:45 +0000 Subject: [PATCH] reduce diff --- tests/generation/test_utils.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tests/generation/test_utils.py b/tests/generation/test_utils.py index d29accc93c590a..0ed45cfd04b06b 100644 --- a/tests/generation/test_utils.py +++ b/tests/generation/test_utils.py @@ -4206,6 +4206,7 @@ def test_padding_input_contrastive_search_t5(self): self.assertEqual(generated_text_no_padding, generated_text_with_padding) self.assertEqual(generated_text_no_padding, "Ich muss diese Aufgabe vor Ende des Tages beenden.") + @require_torch class UtilsFunctionsTest(unittest.TestCase): def test_speculative_sampling(self):