Skip to content

Commit ccf2ca1

Browse files
authored
skip some test_sdpa_can_dispatch_on_flash (#39092)
* fix * fix * fix --------- Co-authored-by: ydshieh <[email protected]>
1 parent a11f692 commit ccf2ca1

File tree

1 file changed

+18
-1
lines changed

1 file changed

+18
-1
lines changed

tests/test_modeling_common.py

Lines changed: 18 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3748,7 +3748,24 @@ def test_sdpa_can_dispatch_on_flash(self):
37483748
self.skipTest(
37493749
"PaliGemma-like models currently (transformers==4.41.0) requires an attention_mask input"
37503750
)
3751-
if config.model_type in ["modernbert", "gemma3", "t5gemma"]:
3751+
if config.model_type in [
3752+
"modernbert",
3753+
"gemma3",
3754+
"t5gemma",
3755+
"diffllama",
3756+
"dpr",
3757+
"eomt",
3758+
"gpt_bigcode",
3759+
"jamba",
3760+
"kosmos-2",
3761+
"mllama",
3762+
"pixtral",
3763+
"sam",
3764+
"sam_hq",
3765+
"zamba2",
3766+
"sam_vision_model",
3767+
"sam_hq_vision_model",
3768+
]:
37523769
self.skipTest(
37533770
reason=f"{config.model_type} currently (transformers==4.52.0) automatically adds an attention_mask input"
37543771
)

0 commit comments

Comments
 (0)