transformers
Bug Fix: Dynamically set return_lse flag in FlexAttention
#40352

Merged

Bug Fix: Dynamically set return_lse flag in FlexAttention #40352

vasqu merged 5 commits into huggingface:main from amd-lalithnc:main

bug fix - return_lse dynamically set

6a91a9dc

vasqu commented on 2025-08-21

addressed compatibility with return type - flex_attention_forward

2350c9fb

rename variables

b6baacfe

Merge branch 'main' into main

2a053e7f

vasqu approved these changes on 2025-08-22

revert changes to commits

1b69c016

ArthurZucker approved these changes on 2025-08-22

vasqu enabled auto-merge (squash) 114 days ago

vasqu merged 0a21e870 into main 114 days ago

Reviewers

vasqu

ArthurZucker

Assignees

No one assigned

Labels

None yet

Milestone

No milestone