transformers/tests/models/whisper
Joao Gante 7130a22db9
Generate: consistently handle special tokens as tensors (#30624)
* tmp commit

* [test_all] mvp

* missing not

* [test_all] final test fixes

* fix musicgen_melody and rag

* [test_all] empty commit

* PR comments

* Update src/transformers/generation/utils.py

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

---------

Co-authored-by: Arthur <48595927+ArthurZucker@users.noreply.github.com>
2024-05-09 18:01:57 +01:00
..
__init__.py Add WhisperModel to transformers (#19166) 2022-10-05 22:28:31 +02:00
test_feature_extraction_whisper.py [Feature Extractors] Fix kwargs to pre-trained (#30260) 2024-04-19 11:16:08 +01: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 Generate: consistently handle special tokens as tensors (#30624) 2024-05-09 18:01:57 +01: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