user avatar
Add head_mask and decoder_head_mask to PyTorch LED (#9856)
Daniel Stancl authored
* 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
Name Last commit Last update