transformers/tests/models
Arthur f60eec4003
update relative positional embedding (#20203)
* update relative positional embedding

* make fix copies

* add `use_cache` to list of arguments

* fixup

* 1line fucntion

* add `test_decoder_model_past_with_large_inputs_relative_pos_emb`

* add relative pos embedding test for more models

* style
2022-11-15 10:46:34 +01:00
..
albert Improve model tester (#19984) 2022-11-02 17:38:44 +01:00
auto Fix tapas scatter (#20149) 2022-11-14 01:04:26 -05:00
bart Generate: move generation_*.py src files into generation/*.py (#20096) 2022-11-09 15:34:08 +00:00
barthez Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
bartpho Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
beit Add Image Processors (#19796) 2022-11-02 11:57:36 +00:00
bert update relative positional embedding (#20203) 2022-11-15 10:46:34 +01:00
bert_generation Generate: move generation_*.py src files into generation/*.py (#20096) 2022-11-09 15:34:08 +00:00
bert_japanese Add sentencepiece to BertJapaneseTokenizer (#19769) 2022-10-21 10:04:49 -04:00
bertweet Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
big_bird wrap forward passes with torch.no_grad() (#19273) 2022-10-04 16:13:22 +02:00
bigbird_pegasus Generate: move generation_*.py src files into generation/*.py (#20096) 2022-11-09 15:34:08 +00:00
blenderbot Generate: move generation_*.py src files into generation/*.py (#20096) 2022-11-09 15:34:08 +00:00
blenderbot_small Generate: move generation_*.py src files into generation/*.py (#20096) 2022-11-09 15:34:08 +00:00
bloom Generate: move generation_*.py src files into generation/*.py (#20096) 2022-11-09 15:34:08 +00:00
bort Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
byt5 Replace as_target context managers by direct calls (#18325) 2022-07-29 08:09:09 -04:00
camembert Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
canine Improve model tester (#19984) 2022-11-02 17:38:44 +01:00
clip [processor] Add 'model input names' property (#20117) 2022-11-10 19:29:20 +00:00
clipseg Update CLIPSegModelTester (#20134) 2022-11-09 15:21:52 +01:00
codegen Generate: move generation_*.py src files into generation/*.py (#20096) 2022-11-09 15:34:08 +00:00
conditional_detr Generate: move generation_*.py src files into generation/*.py (#20096) 2022-11-09 15:34:08 +00:00
convbert wrap forward passes with torch.no_grad() (#19274) 2022-10-04 16:12:03 +02:00
convnext Add Image Processors (#19796) 2022-11-02 11:57:36 +00:00
cpm Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
ctrl Generate: move generation_*.py src files into generation/*.py (#20096) 2022-11-09 15:34:08 +00: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 fix train_new_from_iterator in the case of byte-level tokenizers (#17549) 2022-06-08 15:30:41 +02:00
deberta_v2 🚨 🚨 🚨 Fix Issue 15003: SentencePiece Tokenizers Not Adding Special Tokens in convert_tokens_to_string (#15775) 2022-11-02 15:45:38 -04:00
decision_transformer Generate: move generation_*.py src files into generation/*.py (#20096) 2022-11-09 15:34:08 +00:00
deformable_detr Generate: move generation_*.py src files into generation/*.py (#20096) 2022-11-09 15:34:08 +00:00
deit Add Image Processors (#19796) 2022-11-02 11:57:36 +00:00
detr Generate: move generation_*.py src files into generation/*.py (#20096) 2022-11-09 15:34:08 +00:00
distilbert Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
dit Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
donut remvoe _create_and_check_torch_fx_tracing in specific test files (#18667) 2022-09-07 16:22:09 +02:00
dpr Replace as_target context managers by direct calls (#18325) 2022-07-29 08:09:09 -04:00
dpt Add Image Processors (#19796) 2022-11-02 11:57:36 +00:00
electra Improve model tester (#19984) 2022-11-02 17:38:44 +01:00
encoder_decoder Generate: move generation_*.py src files into generation/*.py (#20096) 2022-11-09 15:34:08 +00:00
ernie update relative positional embedding (#20203) 2022-11-15 10:46:34 +01:00
esm Fix ESM LM head test (#20045) 2022-11-04 12:45:34 +00:00
flaubert Improve model tester (#19984) 2022-11-02 17:38:44 +01:00
flava Fix type - update any PIL.Image.Resampling (#20172) 2022-11-11 16:55:59 +00:00
fnet wrap forward passes with torch.no_grad() (#19413) 2022-10-10 15:03:46 -04:00
fsmt Generate: move generation_*.py src files into generation/*.py (#20096) 2022-11-09 15:34:08 +00:00
funnel Update serving code to enable saved_model=True (#18153) 2022-07-22 18:05:38 +01:00
glpn Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
gpt_neo Generate: move generation_*.py src files into generation/*.py (#20096) 2022-11-09 15:34:08 +00:00
gpt_neox skip some gpt_neox tests that require 80G RAM (#17923) 2022-07-01 09:04:38 -04:00
gpt_neox_japanese Add support for Japanese GPT-NeoX-based model by ABEJA, Inc. (#18814) 2022-09-14 10:17:40 -04:00
gpt2 Generate: move generation_*.py src files into generation/*.py (#20096) 2022-11-09 15:34:08 +00:00
gptj Generate: move generation_*.py src files into generation/*.py (#20096) 2022-11-09 15:34:08 +00:00
groupvit Allow passing arguments to model testers for CLIP-like models (#20044) 2022-11-04 18:01:41 +01:00
herbert Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
hubert Fix train_step, test_step and tests for CLIP (#18684) 2022-09-09 20:01:02 +01:00
ibert Improve model tester (#19984) 2022-11-02 17:38:44 +01:00
imagegpt Generate: move generation_*.py src files into generation/*.py (#20096) 2022-11-09 15:34:08 +00:00
jukebox Add Jukebox model (replaces #16875) (#17826) 2022-11-10 21:05:27 +01:00
layoutlm TF: TF 2.10 unpin + related onnx test skips (#18995) 2022-09-12 19:30:27 +01:00
layoutlmv2 [processor] Add 'model input names' property (#20117) 2022-11-10 19:29:20 +00:00
layoutlmv3 [processor] Add 'model input names' property (#20117) 2022-11-10 19:29:20 +00:00
layoutxlm Fix tapas scatter (#20149) 2022-11-14 01:04:26 -05:00
led Generate: move generation_*.py src files into generation/*.py (#20096) 2022-11-09 15:34:08 +00:00
levit Add Image Processors (#19796) 2022-11-02 11:57:36 +00:00
lilt Generate: move generation_*.py src files into generation/*.py (#20096) 2022-11-09 15:34:08 +00:00
longformer Improve model tester (#19984) 2022-11-02 17:38:44 +01:00
longt5 Generate: move generation_*.py src files into generation/*.py (#20096) 2022-11-09 15:34:08 +00:00
luke Adding fine-tuning models to LUKE (#18353) 2022-08-01 11:09:47 -04:00
lxmert Update serving code to enable saved_model=True (#18153) 2022-07-22 18:05:38 +01:00
m2m_100 Generate: move generation_*.py src files into generation/*.py (#20096) 2022-11-09 15:34:08 +00:00
marian Generate: move generation_*.py src files into generation/*.py (#20096) 2022-11-09 15:34:08 +00:00
markuplm [processor] Add 'model input names' property (#20117) 2022-11-10 19:29:20 +00:00
maskformer Fix image segmentation pipeline errors, resolve backward compatibility issues (#19768) 2022-10-21 18:09:58 +03:00
mbart Generate: move generation_*.py src files into generation/*.py (#20096) 2022-11-09 15:34:08 +00:00
mbart50 Replace as_target context managers by direct calls (#18325) 2022-07-29 08:09:09 -04:00
mctct [processor] Add 'model input names' property (#20117) 2022-11-10 19:29:20 +00:00
megatron_bert Move test model folders (#17034) 2022-05-03 14:42:02 +02: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 Fix train_step, test_step and tests for CLIP (#18684) 2022-09-09 20:01:02 +01:00
mobilenet_v2 add MobileNetV2 model (#17845) 2022-11-14 01:00:10 -05:00
mobilevit Add Image Processors (#19796) 2022-11-02 11:57:36 +00:00
mpnet Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
mt5 Fix expected loss values in some (m)T5 tests (#18177) 2022-07-18 15:26:21 +02:00
mvp Generate: move generation_*.py src files into generation/*.py (#20096) 2022-11-09 15:34:08 +00:00
nezha Generate: move generation_*.py src files into generation/*.py (#20096) 2022-11-09 15:34:08 +00:00
nllb Replace as_target context managers by direct calls (#18325) 2022-07-29 08:09:09 -04:00
nystromformer Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
openai Generate: move generation_*.py src files into generation/*.py (#20096) 2022-11-09 15:34:08 +00:00
opt Generate: move generation_*.py src files into generation/*.py (#20096) 2022-11-09 15:34:08 +00:00
owlvit [OWL-ViT] Make model consistent with CLIP (#20144) 2022-11-11 11:36:17 +01:00
pegasus Generate: move generation_*.py src files into generation/*.py (#20096) 2022-11-09 15:34:08 +00:00
pegasus_x Generate: move generation_*.py src files into generation/*.py (#20096) 2022-11-09 15:34:08 +00:00
perceiver Update defaults and logic to match old FE (#20065) 2022-11-04 19:14:56 +00:00
phobert Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
plbart Generate: move generation_*.py src files into generation/*.py (#20096) 2022-11-09 15:34:08 +00:00
poolformer Add Image Processors (#19796) 2022-11-02 11:57:36 +00:00
prophetnet Generate: move generation_*.py src files into generation/*.py (#20096) 2022-11-09 15:34:08 +00:00
qdqbert Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
rag Avoid GPU OOM for a TF Rag test (#17638) 2022-06-10 18:50:29 +02:00
realm Black preview (#17217) 2022-05-12 16:25:55 -04:00
reformer Generate: move generation_*.py src files into generation/*.py (#20096) 2022-11-09 15:34:08 +00:00
regnet Run tests if skip condition not met (#18764) 2022-08-30 14:03:28 +02:00
rembert Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
resnet FX support for ConvNext, Wav2Vec2 and ResNet (#19053) 2022-09-16 10:57:41 +02:00
retribert fix retribert's test_torch_encode_plus_sent_to_model (#17231) 2022-05-17 14:33:13 +02:00
roberta update relative positional embedding (#20203) 2022-11-15 10:46:34 +01:00
roc_bert update relative positional embedding (#20203) 2022-11-15 10:46:34 +01:00
roformer wrap forward passes with torch.no_grad() (#19438) 2022-10-10 14:54:54 -04:00
segformer Add Image Processors (#19796) 2022-11-02 11:57:36 +00: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 Generate: move generation_*.py src files into generation/*.py (#20096) 2022-11-09 15:34:08 +00:00
speech_to_text [processor] Add 'model input names' property (#20117) 2022-11-10 19:29:20 +00:00
speech_to_text_2 Generate: move generation_*.py src files into generation/*.py (#20096) 2022-11-09 15:34:08 +00:00
splinter Fix Splinter test (#17854) 2022-06-24 16:26:14 +02:00
squeezebert Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
swin remvoe _create_and_check_torch_fx_tracing in specific test files (#18667) 2022-09-07 16:22:09 +02:00
swinv2 Add swin transformer v2 (#17469) 2022-07-27 11:14:47 -04:00
t5 Generate: move generation_*.py src files into generation/*.py (#20096) 2022-11-09 15:34:08 +00:00
table_transformer Generate: move generation_*.py src files into generation/*.py (#20096) 2022-11-09 15:34:08 +00:00
tapas Fix tapas scatter (#20149) 2022-11-14 01:04:26 -05:00
tapex Replace as_target context managers by direct calls (#18325) 2022-07-29 08:09:09 -04:00
time_series_transformer Add a decorator for flaky tests (#19498) 2022-10-12 14:00:17 -04:00
trajectory_transformer Generate: move generation_*.py src files into generation/*.py (#20096) 2022-11-09 15:34:08 +00:00
transfo_xl Generate: move generation_*.py src files into generation/*.py (#20096) 2022-11-09 15:34:08 +00:00
trocr Generate: move generation_*.py src files into generation/*.py (#20096) 2022-11-09 15:34:08 +00:00
unispeech Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
unispeech_sat Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
van has_attentions - consistent test skipping logic and tf tests (#17495) 2022-06-09 09:50:03 +02:00
videomae Add Image Processors (#19796) 2022-11-02 11:57:36 +00:00
vilt Add Image Processors (#19796) 2022-11-02 11:57:36 +00:00
vision_encoder_decoder Generate: move generation_*.py src files into generation/*.py (#20096) 2022-11-09 15:34:08 +00:00
vision_text_dual_encoder [processor] Add 'model input names' property (#20117) 2022-11-10 19:29:20 +00:00
visual_bert wrap forward passes with torch.no_grad() (#19439) 2022-10-10 14:54:36 -04:00
vit Add Image Processors (#19796) 2022-11-02 11:57:36 +00:00
vit_mae TF: tests for (de)serializable models with resized tokens (#19013) 2022-09-16 16:38:08 +01:00
vit_msn Some fixes regarding auto mappings and test class names (#19923) 2022-10-27 14:38:59 +02:00
wav2vec2 [processor] Add 'model input names' property (#20117) 2022-11-10 19:29:20 +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 [processor] Add 'model input names' property (#20117) 2022-11-10 19:29:20 +00:00
wavlm Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
whisper [processor] Add 'model input names' property (#20117) 2022-11-10 19:29:20 +00:00
x_clip Allow passing arguments to model testers for CLIP-like models (#20044) 2022-11-04 18:01:41 +01:00
xglm Generate: move generation_*.py src files into generation/*.py (#20096) 2022-11-09 15:34:08 +00:00
xlm Generate: move generation_*.py src files into generation/*.py (#20096) 2022-11-09 15:34:08 +00:00
xlm_prophetnet Black preview (#17217) 2022-05-12 16:25:55 -04:00
xlm_roberta Black preview (#17217) 2022-05-12 16:25:55 -04:00
xlm_roberta_xl update relative positional embedding (#20203) 2022-11-15 10:46:34 +01:00
xlnet Generate: move generation_*.py src files into generation/*.py (#20096) 2022-11-09 15:34:08 +00:00
yolos [fix] Add DeformableDetrFeatureExtractor (#19140) 2022-09-22 09:45:24 +02:00
yoso fix train_new_from_iterator in the case of byte-level tokenizers (#17549) 2022-06-08 15:30:41 +02:00
__init__.py Move test model folders (#17034) 2022-05-03 14:42:02 +02:00