transformers/tests/models/whisper
Patrick von Platen af3de8d87c
[Whisper, Bart, MBart] Add Flash Attention 2 (#27203)
* add whisper fa2

* correct

* change all

* correct

* correct

* fix more

* fix more

* fix more

* fix more

* fix more

* fix more

* Apply suggestions from code review

Co-authored-by: amyeroberts <22614925+amyeroberts@users.noreply.github.com>

* fix more

* fix more

* fix more

* fix more

* fix more

---------

Co-authored-by: amyeroberts <22614925+amyeroberts@users.noreply.github.com>
2023-11-01 21:03:01 +01: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 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 [Whisper, Bart, MBart] Add Flash Attention 2 (#27203) 2023-11-01 21:03:01 +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