transformers
Patch GPTNeoX to use adequate FA2 if position_ids is provided
#35318
Merged

Patch GPTNeoX to use adequate FA2 if position_ids is provided #35318

taha-yassine
taha-yassine Patch GPTNeoX to use adequate FA2 if position_ids is provided
a453aab8
ArthurZucker
ArthurZucker approved these changes on 2024-12-23
ArthurZucker ArthurZucker merged 2bb60982 into main 364 days ago
taha-yassine taha-yassine deleted the patch_fa2 branch 21 days ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
Milestone