tests/test_modeling_gpt2.py · 505494a86ff5daa0ced517e7d5aef56cac49b3e5 · chenpangpang / transformers

Igor Shalyminov authored Mar 15, 2021

* GPT2DoubleHeadsModel made parallelizeable

* GPT2DoubleHeadsModel added as parallelizeable onto the GPT2 test suite

505494a8

test_modeling_gpt2.py 28.8 KB