tests/models/gptj/test_modeling_gptj.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_gptj.py 26.1 KB

Replace test_modeling_gptj.py