tests/test_modeling_gpt2.py · afb50c663a5d5623906ead1e87481926467d59fa · chenpangpang / transformers

Fix GPT2DoubleHeadsModel to work with model.generate() (#6601) · afb50c66

LSinev authored Nov 16, 2020

* Fix passing token_type_ids during GPT2DoubleHeadsModel.generate() if used

and for GPT2LMHeadModel too

* Update tests to check token_type_ids usage in GPT2 models

afb50c66

test_modeling_gpt2.py 26.2 KB

Replace test_modeling_gpt2.py