transformers/tests/models/whisper
Yih-Dar 08a194fcd6
Fix slow tests for important models to be compatible with A10 runners (#29905)
* fix mistral and mixtral

* add pdb

* fix mixtral tesst

* fix

* fix mistral ?

* add fix gemma

* fix mistral

* fix

* test

* anoter test

* fix

* fix

* fix mistral tests

* fix them again

* final fixes for mistral

* fix padding right

* fix whipser fa2

* fix

* fix

* fix gemma

* test

* fix llama

* fix

* fix

* fix llama gemma

* add class attribute

* fix CI

* clarify whisper

* compute_capability

* rename names in some comments

* Add   # fmt: skip

* make style

* Update tests/models/mistral/test_modeling_mistral.py

Co-authored-by: Arthur <48595927+ArthurZucker@users.noreply.github.com>

* update

* update

---------

Co-authored-by: Younes Belkada <younesbelkada@gmail.com>
Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
Co-authored-by: Arthur <48595927+ArthurZucker@users.noreply.github.com>
2024-04-09 13:28:54 +02:00
..
__init__.py Add WhisperModel to transformers (#19166) 2022-10-05 22:28:31 +02:00
test_feature_extraction_whisper.py [Whisper] Computing features on GPU in batch mode for whisper feature extractor. (#29900) 2024-04-08 10:36:25 +02: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 slow tests for important models to be compatible with A10 runners (#29905) 2024-04-09 13:28:54 +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