transformers/tests/models/gpt2
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_gpt2.py Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
test_modeling_gpt2.py Generate: contrastive search with full optional outputs (#19963) 2022-11-01 18:15:36 +00:00
test_modeling_tf_gpt2.py TF: TF 2.10 unpin + related onnx test skips (#18995) 2022-09-12 19:30:27 +01:00
test_tokenization_gpt2.py Fixing OPT fast tokenizer option. (#18753) 2022-09-15 17:12:58 +02:00