riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-31 02:02:21 +06:00
Code Issues Actions 23 Packages Projects Releases Wiki Activity
c8545d2a9c
transformers/tests/models/whisper
History
bofeng huang c8545d2a9c
[Whisper] Add SpecAugment (#21298) 
* Return and rescale attention_mask

* Add SpecAugment to Whisper modeling

* Fix test

* Update docstring

* Add SpecAug related parameters to model config

* Add the _mask_input_features function to doc

* Fix quality

* Apply suggestions from code review

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

* Remove dev comments

* Add test

* Resolve conflict

* feat: mask {feature, time} prob fast tests

* Apply suggestions from code review

Co-authored-by: Sylvain Gugger <35901082+sgugger@users.noreply.github.com>

---------

Co-authored-by: Arthur <48595927+ArthurZucker@users.noreply.github.com>
Co-authored-by: sanchit-gandhi <sanchit@huggingface.co>
Co-authored-by: Sylvain Gugger <35901082+sgugger@users.noreply.github.com>
2023-02-24 11:07:52 +01:00
..
__init__.py Add WhisperModel to transformers (#19166) 2022-10-05 22:28:31 +02:00
test_feature_extraction_whisper.py Remove CLI spams with Whisper FeatureExtractor (#21267) 2023-02-10 09:15:16 -05:00
test_modeling_flax_whisper.py Fix quality 2023-02-20 03:27:09 -05:00
test_modeling_tf_whisper.py Add WhisperTokenizerFast (#21222) 2023-02-21 06:58:54 +01:00
test_modeling_whisper.py [Whisper] Add SpecAugment (#21298) 2023-02-24 11:07:52 +01:00
test_processor_whisper.py [Whisper] Fix forced decoder ids (#20652) 2022-12-07 16:44:13 +00:00
test_tokenization_whisper.py Add WhisperTokenizerFast (#21222) 2023-02-21 06:58:54 +01:00