transformers
e201864b - [`GPTNeoX`] Fix GPTNeoX + Flash Attention 2 issue (#28645)

Commit

1 year ago

[`GPTNeoX`] Fix GPTNeoX + Flash Attention 2 issue (#28645) Update modeling_gpt_neox.py

References

Author

younesbelkada

Parents