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

…

test_modeling_flax_gpt2.py

…

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