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):
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 )
You can’t perform that action at this time.
0 commit comments