transformers/tests/models/gpt2
Joao Gante e6d27ca5c8
TF: XLA beam search + most generation-compatible models are now also XLA-generate-compatible (#17857)
* working beam search 🎉

* XLA generation compatible with ALL classes

* add xla generation slow test
2022-06-29 12:41:01 +01: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 fix train_new_from_iterator in the case of byte-level tokenizers (#17549) 2022-06-08 15:30:41 +02:00
test_modeling_tf_gpt2.py TF: XLA beam search + most generation-compatible models are now also XLA-generate-compatible (#17857) 2022-06-29 12:41:01 +01:00
test_tokenization_gpt2.py Add OPT (#17088) 2022-05-12 12:24:35 +02:00