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 <ydshieh@users.noreply.github.com>
This commit is contained in:
Yih-Dar
2024-10-11 15:24:01 +02:00
committed by GitHub
parent 1c66be8062
commit 7b06473b8f
2 changed files with 9 additions and 0 deletions

View File

@@ -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)