tests/test_modeling_tf_led.py · 131e258411179d84358f4b28c3ed90fd65f8ba4a · chenpangpang / transformers

Fix TF T5/LED missing cross attn in retrun values (#15511) · 131e2584

Yih-Dar authored Feb 07, 2022



* add cross attn to outputs

* add cross attn to outputs for TFLED

* add undo padding

* remove unused import

* fix style
Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>

131e2584

test_modeling_tf_led.py 23.9 KB

Replace test_modeling_tf_led.py