tests/test_modeling_gpt2.py · 121dd4332b7e44932b0fbe2fa18bc9fa0131402c · chenpangpang / transformers

Jonathan Chang authored Oct 14, 2020



* Add support for gpt2 batch inferencing

* add test

* remove typo
Co-authored-by: patrickvonplaten <patrick.v.platen@gmail.com>

121dd433

test_modeling_gpt2.py 21.2 KB