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:
@@ -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)
|
||||
|
||||
Reference in New Issue
Block a user