Jonathan Chang d160782a53 Add template for adding flax models (#12441) ... * Add option to add flax * Add flax template for __init__.py * Add flax template for .rst * Copy TF modeling template * Add a missing line in modeling_tf_... template * Update first half of modeling_flax_.. * Update encoder flax template * Copy test_modeling_tf... as test_modeling_flax... * Replace some TF to Flax in test_modeling_flax_... * Replace tf to np some function might not work, like _assert_tensors_equal * Replace remaining tf to np (might not work) * Fix cookiecutter * Add Flax in to_replace_... template * Update transformers-cli add-new-model * Save generate_flax in configuration.json This will be read by transformers-cli * Fix to_replace_... and cli * Fix replace cli * Fix cookiecutter name * Move docstring earlier to avoid not defined error * Fix a missing Module * Add encoder-decoder flax template from bart * Fix flax test * Make style * Fix endif * Fix replace all "utf-8 -> unp-8" * Update comment * Fix flax template (add missing ..._DOCSTRING) * Use flax_bart imports in template (was t5) * Fix unp * Update templates/adding_a_new_model/tests * Revert "Fix unp" This reverts commit dc9002a41d. * Remove one line of copied from to suppress CI error * Use generate_tensorflow_pytorch_and_flax * Add a missing part * fix typo * fix flax config * add examples for flax * small rename * correct modeling imports * correct auto loading * corrects some flax tests * correct small typo * correct as type * finish modif * correct more templates * final fixes * add file testers * up * make sure tests match template regex * correct pytorch * correct tf * correct more tf * correct imports * minor error * minor error * correct init * more fixes * correct more flax tests * correct flax test * more fixes * correct docs * update * fix Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com>		2021-09-01 09:49:03 +02:00
..
encoder-bert-tokenizer.json	Add template for adding flax models (#12441)	2021-09-01 09:49:03 +02:00
flax-encoder-bert-tokenizer.json	Add template for adding flax models (#12441)	2021-09-01 09:49:03 +02:00
flax-seq-2-seq-bart-tokenizer.json	Add template for adding flax models (#12441)	2021-09-01 09:49:03 +02:00
pt-encoder-bert-tokenizer.json	Add template for adding flax models (#12441)	2021-09-01 09:49:03 +02:00
pt-seq-2-seq-bart-tokenizer.json	Add template for adding flax models (#12441)	2021-09-01 09:49:03 +02:00
standalone.json	Add template for adding flax models (#12441)	2021-09-01 09:49:03 +02:00
tf-encoder-bert-tokenizer.json	Add template for adding flax models (#12441)	2021-09-01 09:49:03 +02:00
tf-seq-2-seq-bart-tokenizer.json	Add template for adding flax models (#12441)	2021-09-01 09:49:03 +02:00