Bug Fix: Dynamically set return_lse flag in FlexAttention#40352

vasqu merged 5 commits intohuggingface:mainhuggingface/transformers:mainfrom

amd-lalithnc:mainamd-lalithnc/transformers:mainCopy head branch name to clipboard