tests/test_modeling_tf_marian.py · 38a716cd41f22f6a7d5ff3dc081903090198803a · chenpangpang / transformers

TF BART models - Add `cross_attentions` to model output and fix... · 38a716cd

Daniel Stancl authored Apr 26, 2021

TF BART models - Add `cross_attentions` to model output and fix cross-attention head masking (#10699)

* Add cross_attn_head_mask to BART

* Fix cross_attentions in TFBart-like models

* This commit enables returning of `cross_attentions`
for TFBart-like models

* It also fixes attention head masking in cross-attenion module

* Update TF model templates

* Fix missing , in TF model templates

* Fix typo: congig -> config

38a716cd

test_modeling_tf_marian.py 17.2 KB

Replace test_modeling_tf_marian.py