transformers/tests/models/whisper
Hz, Ji 50378cbf6c
device agnostic models testing (#27146)
* device agnostic models testing

* add decorator `require_torch_fp16`

* make style

* apply review suggestion

* Oops, the fp16 decorator was misused
2023-10-31 18:12:14 +01:00
..
__init__.py
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 Make Whisper Encoder's sinusoidal PE non-trainable by default (#26032) 2023-10-11 09:08:54 +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 device agnostic models testing (#27146) 2023-10-31 18:12:14 +01:00
test_processor_whisper.py feat: Whisper prompting (#22496) 2023-05-19 09:33:11 +01:00
test_tokenization_whisper.py Copied from for test files (#26713) 2023-10-11 14:12:09 +02:00