transformers/tests/models/t5
Joao Gante 831590f6a9
Generate: contrastive search with full optional outputs (#19963)
* Use beam search functionality; Add extra outputs and test

* Add full tests for contrastive search

* Add error message on unconventional cache format
2022-11-01 18:15:36 +00:00
..
__init__.py Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
test_modeling_flax_t5.py Flax t5 Encoder (#17784) 2022-06-30 00:49:02 +02:00
test_modeling_t5.py Generate: contrastive search with full optional outputs (#19963) 2022-11-01 18:15:36 +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 Replace as_target context managers by direct calls (#18325) 2022-07-29 08:09:09 -04:00