transformers/tests/models
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
..
albert Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
altclip Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
audio_spectrogram_transformer Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
auto [tokenizer] sanitize saved config (#21483) 2023-02-07 10:51:45 -08:00
bart Apply ruff flake8-comprehensions (#21694) 2023-02-22 09:14:54 +01:00
barthez Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
bartpho Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
beit Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
bert Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
bert_generation Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
bert_japanese Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
bertweet Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
big_bird Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
bigbird_pegasus Generate: move generation_*.py src files into generation/*.py (#20096) 2022-11-09 15:34:08 +00:00
biogpt Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
bit [Backbones] Improve out features (#20675) 2022-12-09 09:14:52 +01:00
blenderbot Apply ruff flake8-comprehensions (#21694) 2023-02-22 09:14:54 +01:00
blenderbot_small Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
blip [BLIP] update blip path on slow tests (#21476) 2023-02-17 18:26:36 +00:00
blip_2 Fix Blip-2 CI again (#21637) 2023-02-15 10:59:42 +01:00
bloom Apply ruff flake8-comprehensions (#21694) 2023-02-22 09:14:54 +01:00
bort Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
bridgetower Tag tests as slow (#21537) 2023-02-09 14:46:15 +00:00
byt5 Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
camembert Added with torch.no_grad() to Camembert integration test (#21544) 2023-02-10 10:58:29 +01:00
canine Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
chinese_clip Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
clap [CLAP] Fix few broken things (#21670) 2023-02-17 11:32:14 +01:00
clip Apply ruff flake8-comprehensions (#21694) 2023-02-22 09:14:54 +01:00
clipseg Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
codegen Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
conditional_detr Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
convbert Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
convnext Add EfficientNet (#21563) 2023-02-20 16:37:11 +03:00
cpm Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
ctrl Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
cvt [CvT] Tensorflow implementation (#18597) 2022-10-11 18:16:52 +01:00
data2vec Apply ruff flake8-comprehensions (#21694) 2023-02-22 09:14:54 +01:00
deberta Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
deberta_v2 Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
decision_transformer Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
deformable_detr Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
deit Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
deta 🚨🚨🚨 Enforce single model initialization (#21431) 2023-02-09 15:46:26 -05:00
detr Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
dinat Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
distilbert Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
dit Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
donut Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
dpr Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
dpt 🚨🚨🚨 Enforce single model initialization (#21431) 2023-02-09 15:46:26 -05:00
efficientformer Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
efficientnet Add EfficientNet (#21563) 2023-02-20 16:37:11 +03:00
electra Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
encoder_decoder Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
ernie update relative positional embedding (#20203) 2022-11-15 10:46:34 +01:00
ernie_m Add Ernie-M Model to huggingface (#21349) 2023-02-15 09:24:56 -05:00
esm Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
flaubert Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
flava Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
fnet Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
fsmt Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
funnel Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
git Generate: Fix GIT batched captioning (#21738) 2023-02-23 09:50:37 +00:00
glpn Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
gpt_neo Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
gpt_neox Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
gpt_neox_japanese Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
gpt_sw3 Add gpt-sw3 model to transformers (#20209) 2022-12-12 13:12:13 -05:00
gpt2 [OPT] Adds GPT2TokenizerFast to the list of tokenizer to use for OPT. (#20823) 2023-02-07 17:35:28 +01:00
gptj [CI ] Remove past in favor of pat_key_values (#21443) 2023-02-07 09:51:35 +01:00
gptsan_japanese Fix GPTSanJapaneseModel (#21731) 2023-02-22 11:09:04 +01:00
graphormer Fix Graphormer test suite (#21419) 2023-02-02 16:29:13 +01:00
groupvit Apply ruff flake8-comprehensions (#21694) 2023-02-22 09:14:54 +01:00
herbert Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
hubert Skipping more high mem tests - Wav2Vec2 Hubert (#21647) 2023-02-15 16:00:50 +00:00
ibert Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
imagegpt Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
jukebox Apply ruff flake8-comprehensions (#21694) 2023-02-22 09:14:54 +01:00
layoutlm Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
layoutlmv2 Apply ruff flake8-comprehensions (#21694) 2023-02-22 09:14:54 +01:00
layoutlmv3 Apply ruff flake8-comprehensions (#21694) 2023-02-22 09:14:54 +01:00
layoutxlm Apply ruff flake8-comprehensions (#21694) 2023-02-22 09:14:54 +01:00
led Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
levit Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
lilt Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
longformer Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
longt5 Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
luke Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
lxmert Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
m2m_100 Exclude the madeup words from M2M100Tokenizer.vocab_size (#20976) 2023-02-08 09:19:06 -05:00
marian Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
markuplm Apply ruff flake8-comprehensions (#21694) 2023-02-22 09:14:54 +01:00
mask2former Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
maskformer Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
mbart Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
mbart50 Replace as_target context managers by direct calls (#18325) 2022-07-29 08:09:09 -04:00
mctct Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
megatron_bert Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
megatron_gpt2 Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
mluke Black preview (#17217) 2022-05-12 16:25:55 -04:00
mobilebert Tag tests as slow (#21537) 2023-02-09 14:46:15 +00:00
mobilenet_v1 Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
mobilenet_v2 Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
mobilevit Apply ruff flake8-comprehensions (#21694) 2023-02-22 09:14:54 +01:00
mpnet Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
mt5 Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
mvp Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
nat Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
nezha Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
nllb Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
nystromformer Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
oneformer Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
openai Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
opt Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
owlvit Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
pegasus Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
pegasus_x Generate: move generation_*.py src files into generation/*.py (#20096) 2022-11-09 15:34:08 +00:00
perceiver Apply ruff flake8-comprehensions (#21694) 2023-02-22 09:14:54 +01:00
phobert Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
plbart Generate: fix plbart generation tests (#20391) 2022-11-22 17:56:04 +00:00
poolformer Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
prophetnet 🚨🚨🚨 Enforce single model initialization (#21431) 2023-02-09 15:46:26 -05:00
qdqbert Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
rag Tag tests as slow (#21537) 2023-02-09 14:46:15 +00:00
realm Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
reformer 🚨🚨🚨 Enforce single model initialization (#21431) 2023-02-09 15:46:26 -05:00
regnet [Tests] Improve test_attention_outputs (#20701) 2022-12-14 14:41:40 +01:00
rembert Wrap RemBert integration test forward passes with torch.no_grad() (#21503) 2023-02-08 14:00:52 +01:00
resnet [Tests] Improve test_attention_outputs (#20701) 2022-12-14 14:41:40 +01:00
retribert Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
roberta Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
roberta_prelayernorm Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
roc_bert Apply ruff flake8-comprehensions (#21694) 2023-02-22 09:14:54 +01:00
roformer Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
segformer Apply ruff flake8-comprehensions (#21694) 2023-02-22 09:14:54 +01:00
sew Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
sew_d Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
speech_encoder_decoder Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
speech_to_text Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
speech_to_text_2 Generate: move generation_*.py src files into generation/*.py (#20096) 2022-11-09 15:34:08 +00:00
speecht5 [SpeechT5HifiGan] Handle batched inputs (#21702) 2023-02-22 11:16:56 +01:00
splinter Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
squeezebert Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
swin Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
swin2sr Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
swinv2 Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
switch_transformers Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
t5 Apply ruff flake8-comprehensions (#21694) 2023-02-22 09:14:54 +01:00
table_transformer Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
tapas Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
tapex Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
time_series_transformer Time series transformer: input projection and Std scaler (#21020) 2023-02-22 07:50:13 +01:00
timesformer Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
trajectory_transformer Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
transfo_xl Apply ruff flake8-comprehensions (#21694) 2023-02-22 09:14:54 +01:00
trocr Generate: move generation_*.py src files into generation/*.py (#20096) 2022-11-09 15:34:08 +00:00
tvlt Apply ruff flake8-comprehensions (#21694) 2023-02-22 09:14:54 +01:00
unispeech Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
unispeech_sat Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
upernet Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
van [Tests] Improve test_attention_outputs (#20701) 2022-12-14 14:41:40 +01:00
videomae Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
vilt Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
vision_encoder_decoder Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
vision_text_dual_encoder Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
visual_bert Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
vit Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
vit_hybrid 🚨🚨🚨 Enforce single model initialization (#21431) 2023-02-09 15:46:26 -05:00
vit_mae Apply ruff flake8-comprehensions (#21694) 2023-02-22 09:14:54 +01:00
vit_msn Some fixes regarding auto mappings and test class names (#19923) 2022-10-27 14:38:59 +02:00
wav2vec2 Apply ruff flake8-comprehensions (#21694) 2023-02-22 09:14:54 +01:00
wav2vec2_conformer [Test] Fix W2V-Conformer integration test (#17303) 2022-05-17 18:20:36 +02:00
wav2vec2_phoneme Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
wav2vec2_with_lm Apply ruff flake8-comprehensions (#21694) 2023-02-22 09:14:54 +01:00
wavlm Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
whisper [Whisper] Add SpecAugment (#21298) 2023-02-24 11:07:52 +01:00
x_clip Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
xglm Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
xlm Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
xlm_prophetnet Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
xlm_roberta Added with torch.no_grad() to XLM-Roberta integration test (#21547) 2023-02-09 21:49:54 +01:00
xlm_roberta_xl Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
xlnet Apply ruff flake8-comprehensions (#21694) 2023-02-22 09:14:54 +01:00
xmod [WIP] Move X-MOD models to facebook organization (#21640) 2023-02-16 09:18:25 -05:00
yolos Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
yoso Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
__init__.py Move test model folders (#17034) 2022-05-03 14:42:02 +02:00