From 7b06473b8f6b7c440c65459f3dc0a2f2454c91e7 Mon Sep 17 00:00:00 2001 From: Yih-Dar <2521628+ydshieh@users.noreply.github.com> Date: Fri, 11 Oct 2024 15:24:01 +0200 Subject: [PATCH] avoid many failures for ImageGPT (#34071) * skip * [run-slow] imagegpt * skip * [run-slow] imagegpt * [run-slow] imagegpt,video_llava * skip * [run-slow] imagegpt,video_llava --------- Co-authored-by: ydshieh --- tests/models/imagegpt/test_modeling_imagegpt.py | 6 ++++++ tests/models/video_llava/test_modeling_video_llava.py | 3 +++ 2 files changed, 9 insertions(+) diff --git a/tests/models/imagegpt/test_modeling_imagegpt.py b/tests/models/imagegpt/test_modeling_imagegpt.py index d8ceed6885..0797267552 100644 --- a/tests/models/imagegpt/test_modeling_imagegpt.py +++ b/tests/models/imagegpt/test_modeling_imagegpt.py @@ -257,6 +257,12 @@ class ImageGPTModelTest(ModelTesterMixin, GenerationTesterMixin, PipelineTesterM self.assertEqual(len(scores), length) self.assertListEqual([iter_scores.shape for iter_scores in scores], [expected_shape] * len(scores)) + @unittest.skip( + reason="After #33632, this test still passes, but many subsequential tests fail with `device-side assert triggered`" + ) + def test_beam_search_generate_dict_outputs_use_cache(self): + pass + def setUp(self): self.model_tester = ImageGPTModelTester(self) self.config_tester = ConfigTester(self, config_class=ImageGPTConfig, n_embd=37) diff --git a/tests/models/video_llava/test_modeling_video_llava.py b/tests/models/video_llava/test_modeling_video_llava.py index df8fe0b5dc..492dcb9bae 100644 --- a/tests/models/video_llava/test_modeling_video_llava.py +++ b/tests/models/video_llava/test_modeling_video_llava.py @@ -237,6 +237,9 @@ class VideoLlavaForConditionalGenerationModelTest(ModelTesterMixin, GenerationTe def test_sdpa_can_dispatch_on_flash(self): pass + @unittest.skip( + reason="After #33533, this still passes, but many subsequential tests fail with `device-side assert triggered`" + ) def test_mixed_input(self): config, inputs = self.model_tester.prepare_config_and_inputs_for_common() for model_class in self.all_model_classes: