tests/test_modeling_led.py · 71bdc076dd4ba2f3264283d4bc8617755206dccd · chenpangpang / transformers

Add head_mask and decoder_head_mask to PyTorch LED (#9856) · 71bdc076

Daniel Stancl authored Feb 02, 2021

* Add {decoder_,}head_mask to LED

* Fix create_custom_forward signatue in encoder

* Add head_mask to longformer

* Add head_mask to longformer to fix dependencies
of LED on Longformer.

* Not working yet

* Add mising one input in longofrmer_modeling.py

* make fix-copies

71bdc076

test_modeling_led.py 93.7 KB

Replace test_modeling_led.py