transformers/tests/models/whisper
Younes Belkada d979cf6efd
[Whiper] add get_input_embeddings to WhisperForAudioClassification (#22133)
* add `get_input_embeddings` to `WhisperForAudioClassification`

* add common tests

* fix another common test

* Update tests/models/whisper/test_modeling_whisper.py

Co-authored-by: Arthur <48595927+ArthurZucker@users.noreply.github.com>

* fix style

---------

Co-authored-by: Arthur <48595927+ArthurZucker@users.noreply.github.com>
2023-03-13 19:46:01 +01:00
..
__init__.py Add WhisperModel to transformers (#19166) 2022-10-05 22:28:31 +02:00
test_feature_extraction_whisper.py [Whisper] Add rescaling function with do_normalize (#21263) 2023-03-02 14:17:21 +01:00
test_modeling_flax_whisper.py Fix quality 2023-02-20 03:27:09 -05:00
test_modeling_tf_whisper.py 🔥Rework pipeline testing by removing PipelineTestCaseMeta 🚀 (#21516) 2023-02-28 19:40:57 +01:00
test_modeling_whisper.py [Whiper] add get_input_embeddings to WhisperForAudioClassification (#22133) 2023-03-13 19:46:01 +01:00
test_processor_whisper.py [Whisper] Fix forced decoder ids (#20652) 2022-12-07 16:44:13 +00:00
test_tokenization_whisper.py Refactor whisper asr pipeline to include language too. (#21427) 2023-03-02 18:12:19 +01:00