transformers/tests/models/whisper
Ondřej Cífka bcd42c4af9
Fix kwargs handling in generate_with_fallback (#29225)
* Fix generate_with_fallback **kwargs

* Change pop to get

* Delete keys from kwargs to prevent overriding generation_config

* Revert to passing kwargs by reference, but make a (shallow) copy

* dict -> copy.copy

* Add test_whisper_longform_multi_batch_beam
2024-04-03 17:51:03 +02:00
..
__init__.py Add WhisperModel to transformers (#19166) 2022-10-05 22:28:31 +02:00
test_feature_extraction_whisper.py [Whisper] Use torch for stft if available (#26119) 2023-12-21 11:04:05 +00:00
test_modeling_flax_whisper.py [Styling] stylify using ruff (#27144) 2023-11-16 17:43:19 +01:00
test_modeling_tf_whisper.py More TF fixes (#28081) 2023-12-18 15:26:03 +00:00
test_modeling_whisper.py Fix kwargs handling in generate_with_fallback (#29225) 2024-04-03 17:51:03 +02:00
test_processor_whisper.py feat: Whisper prompting (#22496) 2023-05-19 09:33:11 +01:00
test_tokenization_whisper.py Adds pretrained IDs directly in the tests (#29534) 2024-03-13 14:53:27 +01:00