transformers

Support passing flash_attn_kwargs when gradient_checkpointing is enabled

#37037

Merged