transformers/templates/adding_a_new_model/cookiecutter-template-{{cookiecutter.modelname}}
Yih-Dar dc05dd539f
Fix TF Causal LM models' returned logits (#15256)
* Fix TF Causal LM models' returned logits

* Fix expected shape in the tests

Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
2022-02-01 11:04:07 +00: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 Fix loss calculation in TFXXXForTokenClassification models (#15294) 2022-01-31 11:43:08 -05: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 Happy New Year! (#15094) 2022-01-10 12:05:57 -05:00
tokenization_fast_{{cookiecutter.lowercase_modelname}}.py Happy New Year! (#15094) 2022-01-10 12:05:57 -05:00