transformers/tests/models/whisper
Joao Gante 78cda46f17
Generate: Add assisted generation (#22211)
* working mvp

* remove breakpoint

* fix commit

* standardize outputs

* tmp commit

* tests almost ready

* tmp commit

* skip a few models

* Add streaming; Docs and examples

* document limitations

* PR commits

* Amy PR comments
2023-04-18 17:36:56 +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 Generate: Add assisted generation (#22211) 2023-04-18 17:36:56 +01:00
test_modeling_whisper.py Generate: Add assisted generation (#22211) 2023-04-18 17:36:56 +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