transformers

0a21e870 - Bug Fix: Dynamically set return_lse flag in FlexAttention (#40352)

Bug Fix: Dynamically set return_lse flag in FlexAttention (#40352) * bug fix - return_lse dynamically set * addressed compatibility with return type - flex_attention_forward * rename variables * revert changes to commits