transformers/tests/models/whisper
Patrick von Platen 4151fbb49c
[Whisper] Add sequential longform decoding (#27492)
* [Whisper] Add seq gen

* [Whisper] Add seq gen

* more debug

* Fix whisper logit processor

* Improve whisper code further

* Fix more

* more debug

* more debug

* Improve further

* Add tests

* Prep for batch size > 1

* Get batch_size>1 working

* Correct more

* Add extensive tests

* more debug

* more debug

* more debug

* add more tests

* more debug

* Apply suggestions from code review

* more debug

* add comments to explain the code better

* add comments to explain the code better

* add comments to explain the code better

* Add more examples

* add comments to explain the code better

* fix more

* add comments to explain the code better

* add comments to explain the code better

* correct

* correct

* finalize

* Apply suggestions from code review

* Apply suggestions from code review
2023-11-22 13:27:34 +01:00
..
__init__.py Add WhisperModel to transformers (#19166) 2022-10-05 22:28:31 +02:00
test_feature_extraction_whisper.py is_batched fix for remaining 2-D numpy arrays (#23309) 2023-05-23 14:37:35 -04:00
test_modeling_flax_whisper.py [Styling] stylify using ruff (#27144) 2023-11-16 17:43:19 +01:00
test_modeling_tf_whisper.py Make Whisper Encoder's sinusoidal PE non-trainable by default (#26032) 2023-10-11 09:08:54 +01:00
test_modeling_whisper.py [Whisper] Add sequential longform decoding (#27492) 2023-11-22 13:27:34 +01:00
test_processor_whisper.py feat: Whisper prompting (#22496) 2023-05-19 09:33:11 +01:00
test_tokenization_whisper.py [Styling] stylify using ruff (#27144) 2023-11-16 17:43:19 +01:00