transformers/templates/adding_a_new_model/cookiecutter-template-{{cookiecutter.modelname}}
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
..
__init__.py Fix model templates (#9999) 2021-02-04 07:47:26 -05:00
{{cookiecutter.lowercase_modelname}}.rst Fix model templates (#9999) 2021-02-04 07:47:26 -05:00
configuration_{{cookiecutter.lowercase_modelname}}.py [PyTorch Bart] Split Bart into different models (#9343) 2021-01-05 22:00:05 +01:00
configuration.json Model Templates for Seq2Seq (#9251) 2020-12-22 23:41:20 +01:00
modeling_{{cookiecutter.lowercase_modelname}}.py Model templates (#10072) 2021-02-08 09:07:02 -05:00
modeling_tf_{{cookiecutter.lowercase_modelname}}.py Making TF BART-like models XLA and AMP compliant (#10191) 2021-02-17 17:48:56 +01:00
test_modeling_{{cookiecutter.lowercase_modelname}}.py Fix model templates (#9999) 2021-02-04 07:47:26 -05:00
test_modeling_tf_{{cookiecutter.lowercase_modelname}}.py Fix TF template (#10189) 2021-02-15 09:21:57 -05:00
to_replace_{{cookiecutter.lowercase_modelname}}.py Fix model templates (#9999) 2021-02-04 07:47:26 -05:00
tokenization_{{cookiecutter.lowercase_modelname}}.py [Tokenizer Utils Base] Make pad function more flexible (#9928) 2021-02-02 10:35:27 +03:00
tokenization_fast_{{cookiecutter.lowercase_modelname}}.py Model Templates for Seq2Seq (#9251) 2020-12-22 23:41:20 +01:00