transformers/tests/models
2023-02-14 14:16:22 +00: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 🚨🚨🚨 Enforce single model initialization (#21431) 2023-02-09 15:46:26 -05: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 Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
blenderbot_small Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
blip Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
blip_2 Generate: input expansion for any model input (#21624) 2023-02-14 14:16:22 +00:00
bloom Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05: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
clip Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05: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 Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05: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 update relative positional embedding (#20203) 2022-11-15 10:46:34 +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
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
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 Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05: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
graphormer Fix Graphormer test suite (#21419) 2023-02-02 16:29:13 +01:00
groupvit Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
herbert Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
hubert Fix TF CTC tests (#21606) 2023-02-13 21:23:00 +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 Ci-jukebox (#20613) 2022-12-06 16:14:03 +01:00
layoutlm Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
layoutlmv2 🚨🚨🚨 Enforce single model initialization (#21431) 2023-02-09 15:46:26 -05:00
layoutlmv3 Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
layoutxlm Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05: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 [processor] Add 'model input names' property (#20117) 2022-11-10 19:29:20 +00: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 Tag tests as slow (#21537) 2023-02-09 14:46:15 +00: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 Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05: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 fix both failing RoCBert tests (#20469) 2022-11-28 17:08:57 +01:00
roformer Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
segformer Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05: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 Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05: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 Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05: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 Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05: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 Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
trocr Generate: move generation_*.py src files into generation/*.py (#20096) 2022-11-09 15:34:08 +00: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 Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
vit_msn Some fixes regarding auto mappings and test class names (#19923) 2022-10-27 14:38:59 +02:00
wav2vec2 Fix TF CTC tests (#21606) 2023-02-13 21:23:00 +00: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 Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
wavlm Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
whisper Fix env. variable type issue in testing (#21609) 2023-02-13 20:53:26 +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 Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
xmod Add X-MOD (#20939) 2023-02-10 15:32:06 +01: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