transformers/tests/models/bert
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_bert.py Fix generate with inputs_embeds as input (#32493) 2024-08-08 18:44:53 +02:00
test_modeling_flax_bert.py Update all references to canonical models (#29001) 2024-02-16 08:16:58 +01:00
test_modeling_tf_bert.py Speed up TF tests by reducing hidden layer counts (#24595) 2023-06-30 16:30:33 +01:00
test_tokenization_bert_tf.py update ruff version (#30932) 2024-05-22 06:40:15 +02:00
test_tokenization_bert.py Skip tests properly (#31308) 2024-06-26 21:59:08 +01:00