transformers

Fix flaky `test_custom_4d_attention_mask`

#35606

Merged