Matt 2322eb8e2f Update serving signatures and make sure we actually use them (#19034) ... * Override save() to use the serving signature as the default * Replace int32 with int64 in all our serving signatures * Remember one very important line so as not to break every test at once * Dtype fix for TFLED * dtype fix for shift_tokens_right in general * Dtype fixes in mBART and RAG * Fix dtypes for test_unpack_inputs * More dtype fixes * Yet more mBART + RAG dtype fixes * Yet more mBART + RAG dtype fixes * Add a check that the model actually has a serving method		2022-09-15 14:34:22 +01:00
..
__init__.py	Fix template init (#17163)	2022-05-10 15:24:23 -04: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	PyTorch >= 1.7.0 and TensorFlow >= 2.4.0 (#19016)	2022-09-14 07:19:02 -04:00
modeling_flax_{{cookiecutter.lowercase_modelname}}.py	[Flax] Add remat (gradient checkpointing) (#17843)	2022-07-01 18:33:54 +01:00
modeling_tf_{{cookiecutter.lowercase_modelname}}.py	Update serving signatures and make sure we actually use them (#19034)	2022-09-15 14:34:22 +01:00
test_modeling_{{cookiecutter.lowercase_modelname}}.py	Move test model folders (#17034)	2022-05-03 14:42:02 +02:00
test_modeling_flax_{{cookiecutter.lowercase_modelname}}.py	Move test model folders (#17034)	2022-05-03 14:42:02 +02:00
test_modeling_tf_{{cookiecutter.lowercase_modelname}}.py	Move test model folders (#17034)	2022-05-03 14:42:02 +02:00
to_replace_{{cookiecutter.lowercase_modelname}}.py	[WIP] Fix Pyright static type checking by replacing if-else imports with try-except (#16578)	2022-05-09 11:28:53 -04:00
tokenization_{{cookiecutter.lowercase_modelname}}.py	Add missing comment quotes (#17379)	2022-06-29 06:16:36 -04:00
tokenization_fast_{{cookiecutter.lowercase_modelname}}.py	Happy New Year! (#15094)	2022-01-10 12:05:57 -05:00