transformers/tests/models/whisper
Joao Gante be2ac0916a
[generate] shape checks in tests compatible with fixed-length caches (+ some minor fixes) (#35993)
* shape checks compatible with static cache

* add test

* tmp

* manually turn on eager attn when we want to output attn

* typo

* generalize to encoder-decoder models

* force compilation on cpu

* tmp commit

* fix static cache shape checks

* models with odd caches

* fix copies

* shorter cache search loop

* use decoder_past_key_values everywhere

* better test variable names and comments

* signature

* rename _check_outputs into _check_generate_outputs

* add comments

* HybridCache future test note
2025-02-10 17:50:54 +00:00
..
__init__.py Add WhisperModel to transformers (#19166) 2022-10-05 22:28:31 +02:00
test_feature_extraction_whisper.py Fix device mismatch error in Whisper model during feature extraction (#35866) 2025-02-04 12:23:08 +01:00
test_modeling_flax_whisper.py Forbid PretrainedConfig from saving generate parameters; Update deprecations in generate-related code 🧹 (#32659) 2024-08-23 11:12:53 +01:00
test_modeling_tf_whisper.py [generate] shape checks in tests compatible with fixed-length caches (+ some minor fixes) (#35993) 2025-02-10 17:50:54 +00:00
test_modeling_whisper.py [generate] shape checks in tests compatible with fixed-length caches (+ some minor fixes) (#35993) 2025-02-10 17:50:54 +00:00
test_processor_whisper.py feat: Whisper prompting (#22496) 2023-05-19 09:33:11 +01:00
test_tokenization_whisper.py Fix flax whisper tokenizer bug (#33151) 2024-09-12 12:21:59 +01:00