From d2871b29754abd0f72cf42c299bb1c041519f7bc Mon Sep 17 00:00:00 2001 From: Yih-Dar <2521628+ydshieh@users.noreply.github.com> Date: Thu, 17 Aug 2023 15:30:46 +0200 Subject: [PATCH] Skip `test_beam_search_xla_generate_simple` for `T5` (#25566) * fix * fix --------- Co-authored-by: ydshieh --- tests/models/t5/test_modeling_tf_t5.py | 2 ++ tests/models/whisper/test_modeling_tf_whisper.py | 5 +++++ 2 files changed, 7 insertions(+) diff --git a/tests/models/t5/test_modeling_tf_t5.py b/tests/models/t5/test_modeling_tf_t5.py index 201bcdd7ae..ec7488e4c3 100644 --- a/tests/models/t5/test_modeling_tf_t5.py +++ b/tests/models/t5/test_modeling_tf_t5.py @@ -549,6 +549,8 @@ class TFT5GenerationIntegrationTests(unittest.TestCase): self.assertListEqual(expected_output_string, output_strings) + # TODO (ydshieh): undo skip once a fix is done on TF side. + @unittest.skip("Skip for now as TF 2.13 breaks it on GPU") @slow def test_beam_search_xla_generate_simple(self): model = TFT5ForConditionalGeneration.from_pretrained("t5-small") diff --git a/tests/models/whisper/test_modeling_tf_whisper.py b/tests/models/whisper/test_modeling_tf_whisper.py index 1bf5c2ccc2..7fae1e466e 100644 --- a/tests/models/whisper/test_modeling_tf_whisper.py +++ b/tests/models/whisper/test_modeling_tf_whisper.py @@ -268,6 +268,11 @@ class TFWhisperModelTest(TFModelTesterMixin, PipelineTesterMixin, unittest.TestC input_name = "input_features" + # TODO (ydshieh): undo skip once a fix is done on TF side. + @unittest.skip("Skip for now as TF 2.13 breaks it on GPU") + def test_xla_generate_slow(self): + super().test_xla_generate_slow() + def setUp(self): self.model_tester = TFWhisperModelTester(self) self.config_tester = ConfigTester(self, config_class=WhisperConfig)