Skip to content

GitLab

Switch branch/tag
Find file BlameHistoryPermalink
  • Daniel Stancl's avatar
    Fix usage of head masks by TF encoder-decoder models' `generate()` function (#11775) · 0b933584
    Daniel Stancl authored 
    * Fix Bart

* Fix Blenderbot{,_small}

* Fix LED

* Fix Marian

* Fix MBart

* Fix Pegasus

* Fix T5

* Add test for generation with head_mask

* Add a common TF test

* Override a test for the LED model as head masking is not yet properly implemented

* Remove all head_masks from input preparation for LED

* Drop masking for T5 as it needs a bit of refactor
    0b933584
test_modeling_tf_common.py 62.5 KB