transformers/tests/models/bart
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