tests/test_modeling_gpt_neo.py · 2a8115f0838a8fd6de5465658a3afd0868456b58 · chenpangpang / transformers

[WIP] GPT Neo cleanup (#10985) · 2a8115f0

Suraj Patil authored Apr 06, 2021

* better names

* add attention mixin

* all slow tests in one class

* make helper methods static so we can test

* add local attention tests

* better names

* doc

* apply review suggestions

2a8115f0

test_modeling_gpt_neo.py 27.1 KB

Replace test_modeling_gpt_neo.py