transformers/tests/models/bart
Pablo Montalvo 044281605f
Fix generate with inputs_embeds as input (#32493)
* I think inputs_embeds has ndim == 3

* fix sequence length catch

* add generate test

* [run-slow]olmo, persimmon, gemma, gemma2, qwen2, llama

* skip whisper

* fix bart test

* more fixes
2024-08-08 18:44:53 +02:00
..
__init__.py Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
test_modeling_bart.py Fix generate with inputs_embeds as input (#32493) 2024-08-08 18:44:53 +02:00
test_modeling_flax_bart.py Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
test_modeling_tf_bart.py Remove ConversationalPipeline and Conversation object (#31165) 2024-06-07 17:50:18 +01:00
test_tokenization_bart.py Skip tests properly (#31308) 2024-06-26 21:59:08 +01:00