reduce diff

huggingface · Sep 20, 2024 · bf7dc7f · bf7dc7f
1 parent 1e8e794
commit bf7dc7f
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tests/generation/test_utils.py b/tests/generation/test_utils.py
@@ -4206,6 +4206,8 @@ 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):
         # assume vocab size 10, input length 5 + 3 generated candidates
         candidate_input_ids = torch.tensor([[8, 0, 3, 9, 8, 1, 4, 5]])  # input tokens