riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-31 02:02:21 +06:00
Code Issues Actions 33 Packages Projects Releases Wiki Activity
e92190c0f8
transformers/templates/adding_a_new_model/cookiecutter-template-{{cookiecutter.modelname}}
History
Suraj Patil e92190c0f8
Fix Flax params dtype (#13098) 
* fix inits

* fix embed dtype

* fix embed dtype

* add test to check default dtype

* quality

* add type conversion methods for flax models

* more robust casting

* cast sinusoidal positions

* update pegasus

* update albert

* update test

* make sure dtype is passed to every module

* style

* fix electra dense

* fix t5

* quality

* add more tests

* better name

* use the dtype for lm head computation

* fix albert

* style

* fix albert embed dtype

* more tests

* fix vision enc-dec

* cleanup

* fix embed dtype pegasus

* fix default param test

* doc

* update template

* fix final_logits_bias dtype

* Apply suggestions from code review

Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com>

* fix doc

* fix doc

* add detailed docstring for dtype parameter

* remove un-necessary import

Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com>
2021-11-11 14:45:20 +05:30
..
__init__.py Add template for adding flax models (#12441) 2021-09-01 09:49:03 +02:00
{{cookiecutter.lowercase_modelname}}.rst Add template for adding flax models (#12441) 2021-09-01 09:49:03 +02:00
configuration_{{cookiecutter.lowercase_modelname}}.py Make gradient_checkpointing a training argument (#13657) 2021-09-22 07:51:38 -04:00
configuration.json Add template for adding flax models (#12441) 2021-09-01 09:49:03 +02:00
modeling_{{cookiecutter.lowercase_modelname}}.py [Bert2Bert] allow bert2bert + relative embeddings (#14324) 2021-11-09 14:26:58 -05:00
modeling_flax_{{cookiecutter.lowercase_modelname}}.py Fix Flax params dtype (#13098) 2021-11-11 14:45:20 +05:30
modeling_tf_{{cookiecutter.lowercase_modelname}}.py [Docs] More general docstrings (#14028) 2021-10-16 00:48:37 +02:00
test_modeling_{{cookiecutter.lowercase_modelname}}.py Expose get_config() on ModelTesters (#12812) 2021-07-21 04:13:11 -04:00
test_modeling_flax_{{cookiecutter.lowercase_modelname}}.py Add template for adding flax models (#12441) 2021-09-01 09:49:03 +02:00
test_modeling_tf_{{cookiecutter.lowercase_modelname}}.py Add TFEncoderDecoderModel + Add cross-attention to some TF models (#13222) 2021-10-13 00:10:34 +02:00
to_replace_{{cookiecutter.lowercase_modelname}}.py Add template for adding flax models (#12441) 2021-09-01 09:49:03 +02:00
tokenization_{{cookiecutter.lowercase_modelname}}.py Make get_special_tokens_mask consider all tokens (#11163) 2021-04-09 11:57:44 -04:00
tokenization_fast_{{cookiecutter.lowercase_modelname}}.py Copy tokenizer files in each of their repo (#10624) 2021-03-10 11:26:23 -05:00