Skip to content

GitLab

"megatron/git@developer.sourcefind.cn:wuxk1/megatron-lm.git" did not exist on "ee7b19e74e760a43b5179319521c8cc4f55ef72f"
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_led.py 18.4 KB