tests/test_modeling_gptj.py · 41436d3dfb98e0d17f018db29790b65663358edf · chenpangpang / transformers

[GPT-J] Use the `float16` checkpoints in integration tests (#13676) · 7c7d2ec9

Anton Lozhkov authored Sep 22, 2021

* Use fp16 checkpoints

* Style

* Fix outputs and disable OOM tests

* Correct another output

* Use a random smaller model for generation tests

* repo quickfix

* fix gradient checkpointing

7c7d2ec9

test_modeling_gptj.py 23.3 KB

Replace test_modeling_gptj.py