[GPTNeoX] Fix GPTNeoX + Flash Attention 2 issue#28645
Merged
younesbelkada merged 1 commit intomainfrom younesbelkada-patch-1Jan 22, 2024
+1-1
GPTNeoX] Fix GPTNeoX + Flash Attention 2 issue#28645