From ccf2ca162e33f381e454cdb74bf4b41a51ab976d Mon Sep 17 00:00:00 2001 From: Yih-Dar <2521628+ydshieh@users.noreply.github.com> Date: Fri, 27 Jun 2025 23:08:14 +0200 Subject: [PATCH] skip some `test_sdpa_can_dispatch_on_flash` (#39092) * fix * fix * fix --------- Co-authored-by: ydshieh --- tests/test_modeling_common.py | 19 ++++++++++++++++++- 1 file changed, 18 insertions(+), 1 deletion(-) diff --git a/tests/test_modeling_common.py b/tests/test_modeling_common.py index b3625255553..d7a41a6c5d0 100755 --- a/tests/test_modeling_common.py +++ b/tests/test_modeling_common.py @@ -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" )