transformers/templates/adding_a_new_model/cookiecutter-template-{{cookiecutter.modelname}}
Julien Plu 812045adcc
New serving (#9419)
* Add a serving method

* Add albert

* Add serving for BERT and BART

* Add more models

* Finish the serving addition

* Temp fix

* Restore DPR

* Fix funnel attribute

* Fix attributes GPT2

* Fix OpenAIGPT attribute

* Fix T5 attributes

* Fix Bart attributes

* Fix TransfoXL attributes

* Add versioning

* better test

* Update template

* Fix Flaubert

* Fix T5

* Apply style

* Remove unused imports

* Deactivate extra parameters

* Remove too long test + saved_model default to False

* Ignore the saved model test for some models

* Fix some inputs

* Fix mpnet serving

* Trigger CI

* Address all comments
2021-01-07 11:48:49 +01:00
..
__init__.py Model Templates for Seq2Seq (#9251) 2020-12-22 23:41:20 +01:00
{{cookiecutter.lowercase_modelname}}.rst Model Templates for Seq2Seq (#9251) 2020-12-22 23:41:20 +01: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 [GenerationOutputs] Fix GenerationOutputs Tests (#9443) 2021-01-06 19:37:02 +01:00
modeling_tf_{{cookiecutter.lowercase_modelname}}.py New serving (#9419) 2021-01-07 11:48:49 +01:00
test_modeling_{{cookiecutter.lowercase_modelname}}.py [PyTorch Bart] Split Bart into different models (#9343) 2021-01-05 22:00:05 +01:00
test_modeling_tf_{{cookiecutter.lowercase_modelname}}.py [Seq2Seq Templates] Correct some TF-serving errors and add gradient checkpointing to PT by default. (#9334) 2020-12-28 17:51:04 +01:00
to_replace_{{cookiecutter.lowercase_modelname}}.py [Seq2Seq Templates] Fix check_repo.py templates file (#9277) 2020-12-23 11:40:20 +01:00
tokenization_{{cookiecutter.lowercase_modelname}}.py Model Templates for Seq2Seq (#9251) 2020-12-22 23:41:20 +01:00
tokenization_fast_{{cookiecutter.lowercase_modelname}}.py Model Templates for Seq2Seq (#9251) 2020-12-22 23:41:20 +01:00