transformers/templates/adding_a_new_model/cookiecutter-template-{{cookiecutter.modelname}}
Yih-Dar 6a5472a8e1
Force use_cache to be False in PyTorch (#15385)
* use_cache = False for PT models if labels is passed

* Fix for BigBirdPegasusForConditionalGeneration

* add warning if users specify use_cache=True

* Use logger.warning instead of warnings.warn

Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
2022-02-08 16:20:53 +01:00
..
__init__.py Adds missing module_specs for usages of _LazyModule (#15230) 2022-01-21 07:30:12 -05:00
{{cookiecutter.lowercase_modelname}}.mdx Check the repo consistency in model templates test (#15141) 2022-01-14 04:52:38 -05:00
configuration_{{cookiecutter.lowercase_modelname}}.py Happy New Year! (#15094) 2022-01-10 12:05:57 -05:00
configuration.json Add template for adding flax models (#12441) 2021-09-01 09:49:03 +02:00
modeling_{{cookiecutter.lowercase_modelname}}.py Force use_cache to be False in PyTorch (#15385) 2022-02-08 16:20:53 +01:00
modeling_flax_{{cookiecutter.lowercase_modelname}}.py [Fix doc example] fix missing import jnp (#15291) 2022-01-24 14:54:23 +01:00
modeling_tf_{{cookiecutter.lowercase_modelname}}.py Fix TF Causal LM models' returned logits (#15256) 2022-02-01 11:04:07 +00:00
test_modeling_{{cookiecutter.lowercase_modelname}}.py Happy New Year! (#15094) 2022-01-10 12:05:57 -05:00
test_modeling_flax_{{cookiecutter.lowercase_modelname}}.py Happy New Year! (#15094) 2022-01-10 12:05:57 -05:00
test_modeling_tf_{{cookiecutter.lowercase_modelname}}.py Happy New Year! (#15094) 2022-01-10 12:05:57 -05:00
to_replace_{{cookiecutter.lowercase_modelname}}.py Happy New Year! (#15094) 2022-01-10 12:05:57 -05:00
tokenization_{{cookiecutter.lowercase_modelname}}.py fix the tokenizer_config.json file for the slow tokenizer when a fast version is available (#15319) 2022-02-01 16:48:25 +01:00
tokenization_fast_{{cookiecutter.lowercase_modelname}}.py Happy New Year! (#15094) 2022-01-10 12:05:57 -05:00