skip some test_sdpa_can_dispatch_on_flash (#39092)

* fix

* fix

* fix

---------

Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
This commit is contained in:
Yih-Dar
2025-06-27 23:08:14 +02:00
committed by GitHub
parent a11f692895
commit ccf2ca162e

View File

@@ -3748,7 +3748,24 @@ class ModelTesterMixin:
self.skipTest(
"PaliGemma-like models currently (transformers==4.41.0) requires an attention_mask input"
)
if config.model_type in ["modernbert", "gemma3", "t5gemma"]:
if config.model_type in [
"modernbert",
"gemma3",
"t5gemma",
"diffllama",
"dpr",
"eomt",
"gpt_bigcode",
"jamba",
"kosmos-2",
"mllama",
"pixtral",
"sam",
"sam_hq",
"zamba2",
"sam_vision_model",
"sam_hq_vision_model",
]:
self.skipTest(
reason=f"{config.model_type} currently (transformers==4.52.0) automatically adds an attention_mask input"
)