tests/models/bart/test_modeling_bart.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_bart.py 91.7 KB

Replace test_modeling_bart.py