tests/models/t5/test_modeling_t5.py · 831590f6a9d7782f7f37a46a52f2c46e4fd429ad · chenpangpang / transformers

Generate: contrastive search with full optional outputs (#19963) · 831590f6

Joao Gante authored Nov 01, 2022

* Use beam search functionality; Add extra outputs and test

* Add full tests for contrastive search

* Add error message on unconventional cache format

831590f6

test_modeling_t5.py 66.1 KB

Replace test_modeling_t5.py