Skip to content

Commit

Permalink
[WIP] GPT Neo cleanup (#10985)
Browse files Browse the repository at this point in the history 
* 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
  • Loading branch information
@patil-suraj
patil-suraj authored Apr 6, 2021
1 parent 76800fb commit 2a8115f
Show file tree
Hide file tree
Showing 2 changed files with 392 additions and 216 deletions.
Loading

0 comments on commit 2a8115f

Please sign in to comment.