tests/generation/test_generation_utils.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_generation_utils.py 127 KB

Replace test_generation_utils.py