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_bart.py

fix train_new_from_iterator in the case of byte-level tokenizers (#17549)

2022-06-08 15:30:41 +02:00

test_modeling_flax_bart.py

Black preview (#17217)

2022-05-12 16:25:55 -04:00

test_modeling_tf_bart.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_bart.py

…