Remove @slow for test_eager_matches_sdpa_inference (#34558)
* update * update * update * update * update * update * update * update * update * update * update --------- Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
This commit is contained in:
@@ -134,7 +134,7 @@ class IdeficsModelTester:
|
||||
num_attention_heads=self.vision_num_attention_heads,
|
||||
num_hidden_layers=self.vision_num_hidden_layers,
|
||||
intermediate_size=self.vision_intermediate_size,
|
||||
)
|
||||
).to_dict()
|
||||
|
||||
self.perceiver_qk_layer_norms_perceiver = perceiver_qk_layer_norms_perceiver
|
||||
self.perceiver_resampler_depth = perceiver_resampler_depth
|
||||
@@ -316,7 +316,6 @@ class IdeficsModelTester:
|
||||
return floats_tensor([self.batch_size, self.num_channels, self.image_size, self.image_size])
|
||||
|
||||
@require_torch_sdpa
|
||||
@slow
|
||||
@parameterized.expand([("float16",), ("bfloat16",), ("float32",)])
|
||||
def test_eager_matches_sdpa_inference(self, torch_dtype: str):
|
||||
self.skipTest(reason="Idefics has a hard requirement on SDPA, skipping this test")
|
||||
@@ -353,6 +352,12 @@ class IdeficsModelTest(ModelTesterMixin, PipelineTesterMixin, unittest.TestCase)
|
||||
|
||||
return inputs_dict
|
||||
|
||||
@parameterized.expand([("float16",), ("bfloat16",), ("float32",)])
|
||||
@require_torch_sdpa
|
||||
@unittest.skip("Idefics requires both text and image inputs which is currently not done in this test.")
|
||||
def test_eager_matches_sdpa_inference(self):
|
||||
pass
|
||||
|
||||
def test_model_outputs_equivalence(self):
|
||||
try:
|
||||
orig = self.all_model_classes
|
||||
@@ -602,6 +607,12 @@ class IdeficsForVisionText2TextTest(IdeficsModelTest, GenerationTesterMixin, uni
|
||||
)
|
||||
self.config_tester = ConfigTester(self, config_class=IdeficsConfig, hidden_size=37)
|
||||
|
||||
@parameterized.expand([("float16",), ("bfloat16",), ("float32",)])
|
||||
@require_torch_sdpa
|
||||
@unittest.skip("Idefics requires both text and image inputs which is currently not done in this test.")
|
||||
def test_eager_matches_sdpa_inference(self, torch_dtype):
|
||||
pass
|
||||
|
||||
@pytest.mark.generate
|
||||
def test_left_padding_compatibility(self):
|
||||
"""Overwrite because IDEFICS needs image attention mask to be also padded"""
|
||||
|
||||
Reference in New Issue
Block a user