Kamil Akesbi 569743f510 Add sdpa and fa2 the Wav2vec2 family. (#30121) ... * add sdpa to wav2vec. Co-authored-by: kamilakesbi <kamil@huggingface.co> Co-authored-by: jp1924 <jp42maru@gmail.com> * add fa2 to wav2vec2 * add tests * fix attention_mask compatibility with fa2 * minor dtype fix * replace fa2 slow test * fix fa2 slow test * apply code review + add fa2 batch test * add sdpa and fa2 to hubert * sdpa and fa2 to data2vec_audio * sdpa and fa2 to Sew * sdpa to unispeech + unispeech sat * small fix * attention mask in tests Co-authored-by: Sanchit Gandhi <93869735+sanchit-gandhi@users.noreply.github.com> * add_speedup_benchmark_to_doc --------- Co-authored-by: kamil@huggingface.co <kamil.akesbi@gmail.com> Co-authored-by: Sanchit Gandhi <93869735+sanchit-gandhi@users.noreply.github.com>		2024-04-22 18:30:38 +01:00
..
__init__.py	Move test model folders (#17034)	2022-05-03 14:42:02 +02:00
test_feature_extraction_wav2vec2.py	Remove static pretrained maps from the library's internals (#29112)	2024-03-25 10:33:38 +01:00
test_modeling_flax_wav2vec2.py	CI with `num_hidden_layers=2` 🚀🚀🚀 (#25266)	2023-08-02 20:22:36 +02:00
test_modeling_tf_wav2vec2.py	[`Styling`] stylify using ruff (#27144)	2023-11-16 17:43:19 +01:00
test_modeling_wav2vec2.py	Add sdpa and fa2 the Wav2vec2 family. (#30121)	2024-04-22 18:30:38 +01:00
test_processor_wav2vec2.py	[processor] Add 'model input names' property (#20117)	2022-11-10 19:29:20 +00:00
test_tokenization_wav2vec2.py	Fix `skip_special_tokens` for `Wav2Vec2CTCTokenizer._decode` (#29311)	2024-04-02 16:55:11 +02:00