Sanchit Gandhi 485bbe79d5 [Flax] Add remat (gradient checkpointing) (#17843) ... * [Flax] Add remat (gradient checkpointing) * fix variable naming in test * flip: checkpoint using a method * fix naming * fix class naming * apply PVP's suggestions from code review * make fix-copies * fix big-bird, electra, roberta * cookie-cutter * fix flax big-bird * move test to common		2022-07-01 18:33:54 +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	Not use -1e4 as attn mask (#17306)	2022-06-20 16:16:16 +02: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	TF: BART compatible with XLA generation (#17479)	2022-06-20 11:07:46 +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