transformers/tests/models
Younes Belkada 7f4f8b97d0
[Blip2] Fix Blip-2 multi gpu (#21707)
* fix blip multi gpu

* fix

* final changes

* adapt suggestions

* fix failing slow test

* forward contrib credits from testing and suggestions

* reformat

---------

Co-authored-by: akkikiki <akkikiki@users.noreply.github.com>
2023-02-28 17:28:58 +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 Fix the issue of blip model returning loss even when the label is not provided. (#21811) 2023-02-28 09:54:08 -05:00
blip_2 [Blip2] Fix Blip-2 multi gpu (#21707) 2023-02-28 17:28:58 +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 Rename MobileViTModelTest to TFMobileViTModelTest (#21825) 2023-02-28 08:10:29 +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 [SpeechT5] Fix HiFiGAN tests (#21788) 2023-02-24 16:55:38 +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 [T5] Fix torchquant issue (#21843) 2023-02-28 15:09:44 +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] updated expected values for integration test. (#21762) 2023-02-24 12:36:54 +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 Fix-ci-whisper (#21767) 2023-02-24 11:39:25 +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