transformers/tests/models/opt
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 Add OPT (#17088) 2022-05-12 12:24:35 +02:00
test_modeling_flax_opt.py Add final_layer_norm to OPT model (#17785) 2022-06-21 20:26:36 +02:00
test_modeling_opt.py Generate: contrastive search with full optional outputs (#19963) 2022-11-01 18:15:36 +00:00
test_modeling_tf_opt.py add missing require_tf for TFOPTGenerationTest (#19010) 2022-09-13 18:10:11 +02:00