transformers/tests/models/whisper
Matthijs Hollemans 2faa09530b
fix Whisper tests on GPU (#23753)
* move input features to GPU

* skip these tests because undefined behavior

* unskip tests
2023-05-30 09:06:58 -04: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 Temporarily increase tol for PT-FLAX whisper tests (#23288) 2023-05-11 11:43:18 +01:00
test_modeling_tf_whisper.py Better TF docstring types (#23477) 2023-05-24 13:52:52 +01:00
test_modeling_whisper.py fix Whisper tests on GPU (#23753) 2023-05-30 09:06:58 -04:00
test_processor_whisper.py feat: Whisper prompting (#22496) 2023-05-19 09:33:11 +01:00
test_tokenization_whisper.py feat: Whisper prompting (#22496) 2023-05-19 09:33:11 +01:00