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