docs/source/en/model_doc/gpt_neox.md · 9270ab082740a55344a851049a0b69673b6cbdc5 · chenpangpang / transformers

[`Flash Attention 2`] Add flash attention 2 for GPT-Neo-X (#26463) · 9270ab08

Younes Belkada authored Dec 06, 2023

* add flash-attn-2 support for GPT-neo-x

* fixup

* add comment

* revert

* fixes

* update docs

* comment

* again

* fix copies

* add plot + fix copies

* Update docs/source/en/model_doc/gpt_neox.md

9270ab08

gpt_neox.md 5.26 KB

Replace gpt_neox.md