transformers/tests/models/bart
Joao Gante 9fd9476005
[generate] beam search -- fix output cropping (#37080)
* handle jagged beams

* better comment

* bart -- beam search tests print special tokens

* more bart test updates

* more tests!

* better comment
2025-03-28 18:57:51 +01:00
..
__init__.py Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
test_modeling_bart.py [generate] beam search -- fix output cropping (#37080) 2025-03-28 18:57:51 +01:00
test_modeling_flax_bart.py [tests] remove tf/flax tests in /generation (#36235) 2025-02-17 14:59:22 +00:00
test_modeling_tf_bart.py chore: fix typos in the tests directory (#36813) 2025-03-21 10:20:05 +01:00
test_tokenization_bart.py Use lru_cache for tokenization tests (#36818) 2025-03-28 15:09:35 +01:00