tests/test_modeling_tf_mbart.py · 83d803ba02fdd9a1a7f8ed31a5e2910e2835dad0 · chenpangpang / transformers

Making TF BART-like models XLA and AMP compliant (#10191) · 83d803ba

Julien Plu authored Feb 17, 2021

* Update BART

* Update Blenderbot

* Update BlenderbotSmall

* Update Marian

* Update MBart

* Update MBart

* Update Pegasus

* Update template

* Fix Marian and Pegasus

* Apply style

* Default initializer

* Default initializer

* Default initializer

* Remove int32 casts

* Fix template

* Remove more cast

83d803ba

test_modeling_tf_mbart.py 13.8 KB

Replace test_modeling_tf_mbart.py