transformers/tests/models/gptj
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_gptj.py Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
test_modeling_gptj.py Generate: contrastive search with full optional outputs (#19963) 2022-11-01 18:15:36 +00:00
test_modeling_tf_gptj.py TF: GPT-J compatible with XLA generation (#17986) 2022-07-06 15:02:07 +01:00