Add sdpa and fa2 the Wav2vec2 family. #30121
amyeroberts
merged
16 commits into
huggingface:main
from
kamilakesbi:sdpa_fa2_wav2vec
add sdpa to wav2vec.
9c88528a
add fa2 to wav2vec2
ce09b262
add tests
a4711df9
sanchit-gandhi
commented
on 2024-04-08
fix attention_mask compatibility with fa2
02454aff
minor dtype fix
687459b4
replace fa2 slow test
5cb0eb9e
fix fa2 slow test
95f441f0
kamilakesbi
changed the title
apply code review + add fa2 batch test
4a886514
add sdpa and fa2 to hubert
7fb0b3cc
sdpa and fa2 to data2vec_audio
7d02c31a
sdpa and fa2 to Sew
697563f3
sdpa to unispeech + unispeech sat
7c8a8c92
kamilakesbi
changed the title
small fix
89d71b1c
sanchit-gandhi
approved these changes
on 2024-04-11
attention mask in tests
4aa3ee6b
add_speedup_benchmark_to_doc
6338c8a3
Merge branch 'sdpa_fa2_wav2vec' of github.com:kamilakesbi/transformer…
7b277b7e
amyeroberts
approved these changes
on 2024-04-22
Reviewers
Assignees
No one assigned
Labels
None yet
Milestone
No milestone
Login to write a write a comment.
Login via GitHub