File tree Expand file tree Collapse file tree 1 file changed +18
-1
lines changed Expand file tree Collapse file tree 1 file changed +18
-1
lines changed Original file line number Diff line number Diff line change @@ -3748,7 +3748,24 @@ def test_sdpa_can_dispatch_on_flash(self):
3748
3748
self .skipTest (
3749
3749
"PaliGemma-like models currently (transformers==4.41.0) requires an attention_mask input"
3750
3750
)
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
+ ]:
3752
3769
self .skipTest (
3753
3770
reason = f"{ config .model_type } currently (transformers==4.52.0) automatically adds an attention_mask input"
3754
3771
)
You can’t perform that action at this time.
0 commit comments