tests/test_modeling_flax_gptj.py · 4c0dd199c8305903564c2edeae23d294edd4b321 · chenpangpang / transformers

Suraj Patil authored Dec 01, 2021

* add flax gptj

* no bias in attention dense

* no wpe

* fix rotary embeddings

* fix rotary embeds

* fix rotray embeds

* quality

* doc and quality

* fix equivalence tests

4c0dd199

test_modeling_flax_gptj.py 14.2 KB

Replace test_modeling_flax_gptj.py