transformers/tests/models/gpt_neox
Sylvain Gugger fdb120805c
Fix cache for GPT-Neo-X (#17764)
* Fix cache for GPT-Neo-X

* Add more tests
2022-06-20 08:43:36 -04:00
..
__init__.py [WIP] Adding GPT-NeoX-20B (#16659) 2022-05-24 09:31:10 -04:00
test_modeling_gpt_neox.py Fix cache for GPT-Neo-X (#17764) 2022-06-20 08:43:36 -04:00