transformers/tests/models/t5
Lianmin Zheng ac2f6674a3
[FLAX] Add dtype to embedding for bert/bart/opt/t5 (#20340)
* [FLAX] Add dtype to embedding for bert/bart/opt/t5

* Fix all copies

* Add a test case
2022-11-28 10:21:42 -05:00
..
__init__.py Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
test_modeling_flax_t5.py [FLAX] Add dtype to embedding for bert/bart/opt/t5 (#20340) 2022-11-28 10:21:42 -05:00
test_modeling_t5.py Generate: move generation_*.py src files into generation/*.py (#20096) 2022-11-09 15:34:08 +00:00
test_modeling_tf_t5.py 🚨🚨🚨 TF: Remove TFWrappedEmbeddings (breaking: TF embedding initialization updated for encoder-decoder models) (#19263) 2022-10-11 16:48:03 +01:00
test_tokenization_t5.py change the way sentinel tokens can retrived (#20373) 2022-11-23 09:35:44 -05:00