transformers

fix flex attn when optional args aren't passed

#37327

Merged