PR #28645 [`GPTNeoX`] Fix GPTNeoX + Flash Attention 2 issue

[`GPTNeoX`] Fix GPTNeoX + Flash Attention 2 issue #28645

younesbelkada merged 1 commit into main from younesbelkada-patch-1

Update modeling_gpt_neox.py

3f156e40

younesbelkada requested a review from

amyeroberts 2 years ago

amyeroberts approved these changes on 2024-01-22

younesbelkada merged e201864b into main 2 years ago

younesbelkada deleted the younesbelkada-patch-1 branch 2 years ago

Reviewers

amyeroberts

Assignees

No one assigned

Labels

None yet

Milestone

No milestone