transformers/templates
Julien Plu 83d803ba02
Making TF BART-like models XLA and AMP compliant (#10191)
* 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
2021-02-17 17:48:56 +01:00
..
adding_a_new_example_script [examples] make run scripts executable (#10037) 2021-02-05 15:51:18 -08:00
adding_a_new_model Making TF BART-like models XLA and AMP compliant (#10191) 2021-02-17 17:48:56 +01:00