tests/test_modeling_tf_t5.py · 8d79e5ca49ea27ded98de927d220d830f34b7124 · chenpangpang / transformers

Fix head masking for TFT5 (#9877) · 8d79e5ca

Daniel Stancl authored Feb 17, 2021



* Fix head_mask and decoder_head_mask in TFT5 models

* Enable test_headmasking both fot TFT5 tester
and TFT5EncoderOnly tester
Co-authored-by: patrickvonplaten <patrick.v.platen@gmail.com>

8d79e5ca

test_modeling_tf_t5.py 43.5 KB

Replace test_modeling_tf_t5.py