skip some test_sdpa_can_dispatch_on_flash (#39092)
Some checks failed
Self-hosted runner (benchmark) / Benchmark (aws-g5-4xlarge-cache) (push) Has been cancelled
Build documentation / build (push) Has been cancelled
Slow tests on important models (on Push - A10) / Get all modified files (push) Has been cancelled
Self-hosted runner (push-caller) / Check if setup was changed (push) Has been cancelled
Secret Leaks / trufflehog (push) Has been cancelled
Update Transformers metadata / build_and_package (push) Has been cancelled
Slow tests on important models (on Push - A10) / Slow & FA2 tests (push) Has been cancelled
Self-hosted runner (push-caller) / build-docker-containers (push) Has been cancelled
Self-hosted runner (push-caller) / Trigger Push CI (push) Has been cancelled

* 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
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

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