Denisa Roberts 88f50a1e89 Add TensorFlow implementation of EfficientFormer (#22620) ... * Add tf code for efficientformer * Fix return dict bug - return last hidden state after last stage * Fix corresponding return dict bug * Override test tol * Change default values of training to False * Set training to default False X3 * Rm axis from ln * Set init in dense projection * Rm debug stuff * Make style; all tests pass. * Modify year to 2023 * Fix attention biases codes * Update the shape list logic * Add a batch norm eps config * Remove extract comments in test files * Add conditional attn and hidden states return for serving output * Change channel dim checking logic * Add exception for withteacher model in training mode * Revert layer count for now * Add layer count for conditional layer naming * Transpose for conv happens only in main layer * Make tests smaller * Make style * Update doc * Rm from_pt * Change to actual expect image class label * Remove stray print in tests * Update image processor test * Remove the old serving output logic * Make style * Make style * Complete test		2023-05-31 10:43:12 +01:00
..
albert	Better TF docstring types (#23477)	2023-05-24 13:52:52 +01:00
align	Skip failing AlignModelTest::test_multi_gpu_data_parallel_forward (#23374)	2023-05-15 16:46:58 +02:00
altclip	Automatically create/update tiny models (#22275)	2023-03-23 19:14:17 +01:00
audio_spectrogram_transformer	is_batched fix for remaining 2-D numpy arrays (#23309)	2023-05-23 14:37:35 -04:00
auto	Better TF docstring types (#23477)	2023-05-24 13:52:52 +01:00
autoformer	[Time-Series] Autoformer model (#21891)	2023-05-30 10:23:32 +02:00
bart	Better TF docstring types (#23477)	2023-05-24 13:52:52 +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	🔥Rework pipeline testing by removing PipelineTestCaseMeta 🚀 (#21516)	2023-02-28 19:40:57 +01:00
bert	Better TF docstring types (#23477)	2023-05-24 13:52:52 +01:00
bert_generation	🔥Rework pipeline testing by removing PipelineTestCaseMeta 🚀 (#21516)	2023-02-28 19:40:57 +01: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	fix random attention for pytorch's bigbird/pegasus_bigbird (#23056)	2023-05-07 18:55:04 -04:00
bigbird_pegasus	Generate: skip left-padding tests on old models (#23437)	2023-05-18 11:04:51 +01:00
biogpt	Update tiny models and pipeline tests (#23446)	2023-05-18 17:29:04 +02:00
bit	Backbone add mixin tests (#22542)	2023-04-06 13:50:15 +01:00
blenderbot	Better TF docstring types (#23477)	2023-05-24 13:52:52 +01:00
blenderbot_small	Better TF docstring types (#23477)	2023-05-24 13:52:52 +01:00
blip	Better TF docstring types (#23477)	2023-05-24 13:52:52 +01:00
blip_2	[Blip2] skip accelerate test (#22124)	2023-03-13 15:03:21 +01:00
bloom	Generate: add test to check KV format (#23403)	2023-05-16 19:28:19 +01:00
bort	Better TF docstring types (#23477)	2023-05-24 13:52:52 +01:00
bridgetower	Fix a BridgeTower test (#23694)	2023-05-23 17:32:57 +02:00
byt5	Update quality tooling for formatting (#21480)	2023-02-06 18:10:56 -05:00
camembert	Better TF docstring types (#23477)	2023-05-24 13:52:52 +01:00
canine	🔥Rework pipeline testing by removing PipelineTestCaseMeta 🚀 (#21516)	2023-02-28 19:40:57 +01:00
chinese_clip	🔥Rework pipeline testing by removing PipelineTestCaseMeta 🚀 (#21516)	2023-02-28 19:40:57 +01:00
clap	is_batched fix for remaining 2-D numpy arrays (#23309)	2023-05-23 14:37:35 -04:00
clip	Better TF docstring types (#23477)	2023-05-24 13:52:52 +01:00
clipseg	🔥Rework pipeline testing by removing PipelineTestCaseMeta 🚀 (#21516)	2023-02-28 19:40:57 +01:00
codegen	Decorate test_codegen_sample_max_time as flaky (#22953)	2023-04-24 15:27:31 +02:00
conditional_detr	Skip test_multi_gpu_data_parallel_forward for some model tests (#21991)	2023-03-07 14:23:36 +01:00
convbert	Better TF docstring types (#23477)	2023-05-24 13:52:52 +01:00
convnext	Better TF docstring types (#23477)	2023-05-24 13:52:52 +01:00
convnextv2	Update ConvNextV2ModelIntegrationTest::test_inference_image_classification_head (#23402)	2023-05-16 23:35:11 +02:00
cpm	Fix PipelineTests skip conditions (#22320)	2023-03-22 20:02:24 +01:00
cpmant	Update tiny models and pipeline tests (#23446)	2023-05-18 17:29:04 +02:00
ctrl	Better TF docstring types (#23477)	2023-05-24 13:52:52 +01:00
cvt	Better TF docstring types (#23477)	2023-05-24 13:52:52 +01:00
data2vec	Better TF docstring types (#23477)	2023-05-24 13:52:52 +01:00
deberta	Better TF docstring types (#23477)	2023-05-24 13:52:52 +01:00
deberta_v2	Better TF docstring types (#23477)	2023-05-24 13:52:52 +01:00
decision_transformer	🔥Rework pipeline testing by removing PipelineTestCaseMeta 🚀 (#21516)	2023-02-28 19:40:57 +01:00
deformable_detr	🔥Rework pipeline testing by removing PipelineTestCaseMeta 🚀 (#21516)	2023-02-28 19:40:57 +01:00
deit	Better TF docstring types (#23477)	2023-05-24 13:52:52 +01:00
deta	Revert (for now) the change on Deta in #22437 (#22750)	2023-04-13 21:32:29 +02:00
detr	[DETR and friends] Remove is_timm_available (#21814)	2023-03-07 15:19:39 -05:00
dinat	Backbone add mixin tests (#22542)	2023-04-06 13:50:15 +01:00
distilbert	Better TF docstring types (#23477)	2023-05-24 13:52:52 +01:00
dit	Move test model folders (#17034)	2022-05-03 14:42:02 +02:00
donut	🔥Rework pipeline testing by removing PipelineTestCaseMeta 🚀 (#21516)	2023-02-28 19:40:57 +01:00
dpr	Better TF docstring types (#23477)	2023-05-24 13:52:52 +01:00
dpt	🔥Rework pipeline testing by removing PipelineTestCaseMeta 🚀 (#21516)	2023-02-28 19:40:57 +01:00
efficientformer	Add TensorFlow implementation of EfficientFormer (#22620)	2023-05-31 10:43:12 +01:00
efficientnet	Update tiny model summary file for recent models (#22637)	2023-04-06 22:52:59 +02:00
electra	Better TF docstring types (#23477)	2023-05-24 13:52:52 +01:00
encoder_decoder	Better TF docstring types (#23477)	2023-05-24 13:52:52 +01:00
ernie	🔥Rework pipeline testing by removing PipelineTestCaseMeta 🚀 (#21516)	2023-02-28 19:40:57 +01:00
ernie_m	Automatically create/update tiny models (#22275)	2023-03-23 19:14:17 +01:00
esm	Better TF docstring types (#23477)	2023-05-24 13:52:52 +01:00
flaubert	Better TF docstring types (#23477)	2023-05-24 13:52:52 +01:00
flava	🔥Rework pipeline testing by removing PipelineTestCaseMeta 🚀 (#21516)	2023-02-28 19:40:57 +01:00
fnet	Remove some pipeline skip cases (#22865)	2023-04-19 20:27:19 +02:00
focalnet	Update tiny models and pipeline tests (#23446)	2023-05-18 17:29:04 +02:00
fsmt	update_pip_test_mapping (#22606)	2023-04-06 17:56:06 +02:00
funnel	Better TF docstring types (#23477)	2023-05-24 13:52:52 +01:00
git	🔥Rework pipeline testing by removing PipelineTestCaseMeta 🚀 (#21516)	2023-02-28 19:40:57 +01:00
glpn	🔥Rework pipeline testing by removing PipelineTestCaseMeta 🚀 (#21516)	2023-02-28 19:40:57 +01:00
gpt_bigcode	Generate: add test to check KV format (#23403)	2023-05-16 19:28:19 +01:00
gpt_neo	Update tiny models and pipeline tests (#23446)	2023-05-18 17:29:04 +02:00
gpt_neox	Update tiny models and pipeline tests (#23446)	2023-05-18 17:29:04 +02:00
gpt_neox_japanese	🔥Rework pipeline testing by removing PipelineTestCaseMeta 🚀 (#21516)	2023-02-28 19:40:57 +01:00
gpt_sw3	Add gpt-sw3 model to transformers (#20209)	2022-12-12 13:12:13 -05:00
gpt2	Better TF docstring types (#23477)	2023-05-24 13:52:52 +01:00
gptj	Better TF docstring types (#23477)	2023-05-24 13:52:52 +01:00
gptsan_japanese	Update tiny model summary file for recent models (#22637)	2023-04-06 22:52:59 +02:00
graphormer	🔥Rework pipeline testing by removing PipelineTestCaseMeta 🚀 (#21516)	2023-02-28 19:40:57 +01:00
groupvit	Better TF docstring types (#23477)	2023-05-24 13:52:52 +01:00
herbert	Update quality tooling for formatting (#21480)	2023-02-06 18:10:56 -05:00
hubert	Better TF docstring types (#23477)	2023-05-24 13:52:52 +01:00
ibert	🔥Rework pipeline testing by removing PipelineTestCaseMeta 🚀 (#21516)	2023-02-28 19:40:57 +01:00
imagegpt	Generate: skip left-padding tests on old models (#23437)	2023-05-18 11:04:51 +01:00
informer	[Time-Series] Autoformer model (#21891)	2023-05-30 10:23:32 +02:00
jukebox	Update Jukebox tests (#21984)	2023-03-07 04:20:14 +01:00
layoutlm	Better TF docstring types (#23477)	2023-05-24 13:52:52 +01:00
layoutlmv2	Don't use LayoutLMv2 and LayoutLMv3 in some pipeline tests (#22774)	2023-04-17 17:45:20 +02:00
layoutlmv3	Better TF docstring types (#23477)	2023-05-24 13:52:52 +01:00
layoutxlm	Apply ruff flake8-comprehensions (#21694)	2023-02-22 09:14:54 +01:00
led	Better TF docstring types (#23477)	2023-05-24 13:52:52 +01:00
levit	🔥Rework pipeline testing by removing PipelineTestCaseMeta 🚀 (#21516)	2023-02-28 19:40:57 +01:00
lilt	Move is_pipeline_test_to_skip to specific model test classes (#21999)	2023-03-14 10:03:02 +01:00
llama	[LlamaTokenizerFast] nit update post_processor on the fly (#23855)	2023-05-30 16:50:41 +02:00
longformer	Better TF docstring types (#23477)	2023-05-24 13:52:52 +01:00
longt5	update_pip_test_mapping (#22606)	2023-04-06 17:56:06 +02:00
luke	Fix PipelineTests skip conditions (#22320)	2023-03-22 20:02:24 +01:00
lxmert	Better TF docstring types (#23477)	2023-05-24 13:52:52 +01:00
m2m_100	update_pip_test_mapping (#22606)	2023-04-06 17:56:06 +02:00
marian	Better TF docstring types (#23477)	2023-05-24 13:52:52 +01:00
markuplm	Update some MarkupLM tests' expected values (#22667)	2023-04-11 10:00:34 +02:00
mask2former	Fix quality due to ruff release	2023-03-22 20:45:08 -04:00
maskformer	Backbone add mixin tests (#22542)	2023-04-06 13:50:15 +01:00
mbart	Better TF docstring types (#23477)	2023-05-24 13:52:52 +01:00
mbart50	Replace as_target context managers by direct calls (#18325)	2022-07-29 08:09:09 -04:00
mctct	is_batched fix for remaining 2-D numpy arrays (#23309)	2023-05-23 14:37:35 -04:00
mega	Fix MegaModel CI (#22652)	2023-04-07 17:13:04 +02:00
megatron_bert	🔥Rework pipeline testing by removing PipelineTestCaseMeta 🚀 (#21516)	2023-02-28 19:40:57 +01:00
megatron_gpt2	Move test model folders (#17034)	2022-05-03 14:42:02 +02:00
mgp_str	update_pip_test_mapping (#22606)	2023-04-06 17:56:06 +02:00
mluke	Black preview (#17217)	2022-05-12 16:25:55 -04:00
mobilebert	Better TF docstring types (#23477)	2023-05-24 13:52:52 +01:00
mobilenet_v1	🔥Rework pipeline testing by removing PipelineTestCaseMeta 🚀 (#21516)	2023-02-28 19:40:57 +01:00
mobilenet_v2	🔥Rework pipeline testing by removing PipelineTestCaseMeta 🚀 (#21516)	2023-02-28 19:40:57 +01:00
mobilevit	Better TF docstring types (#23477)	2023-05-24 13:52:52 +01:00
mpnet	Better TF docstring types (#23477)	2023-05-24 13:52:52 +01:00
mt5	Better TF docstring types (#23477)	2023-05-24 13:52:52 +01:00
mvp	Generate: skip left-padding tests on old models (#23437)	2023-05-18 11:04:51 +01:00
nat	Backbone add mixin tests (#22542)	2023-04-06 13:50:15 +01:00
nezha	🔥Rework pipeline testing by removing PipelineTestCaseMeta 🚀 (#21516)	2023-02-28 19:40:57 +01:00
nllb	🚨🚨🚨 [NLLB Tokenizer] Fix the prefix tokens 🚨🚨🚨 (#22313)	2023-04-04 14:53:06 +02:00
nllb_moe	tests: Fix flaky test for NLLB-MoE (#22880)	2023-04-21 17:09:40 +01:00
nystromformer	🔥Rework pipeline testing by removing PipelineTestCaseMeta 🚀 (#21516)	2023-02-28 19:40:57 +01:00
oneformer	Automatically create/update tiny models (#22275)	2023-03-23 19:14:17 +01:00
open_llama	add open-llama model with ckpt (#22795)	2023-04-28 11:01:32 -04:00
openai	Better TF docstring types (#23477)	2023-05-24 13:52:52 +01:00
opt	Better TF docstring types (#23477)	2023-05-24 13:52:52 +01:00
owlvit	🔥Rework pipeline testing by removing PipelineTestCaseMeta 🚀 (#21516)	2023-02-28 19:40:57 +01:00
pegasus	Better TF docstring types (#23477)	2023-05-24 13:52:52 +01:00
pegasus_x	update_pip_test_mapping (#22606)	2023-04-06 17:56:06 +02:00
perceiver	🔥Rework pipeline testing by removing PipelineTestCaseMeta 🚀 (#21516)	2023-02-28 19:40:57 +01:00
phobert	Update quality tooling for formatting (#21480)	2023-02-06 18:10:56 -05:00
pix2struct	Update test_batched_inference_image_captioning_conditioned (#23391)	2023-05-16 14:49:24 +02:00
plbart	Generate: skip left-padding tests on old models (#23437)	2023-05-18 11:04:51 +01:00
poolformer	🔥Rework pipeline testing by removing PipelineTestCaseMeta 🚀 (#21516)	2023-02-28 19:40:57 +01:00
prophetnet	Generate: skip left-padding tests on old models (#23437)	2023-05-18 11:04:51 +01:00
qdqbert	🔥Rework pipeline testing by removing PipelineTestCaseMeta 🚀 (#21516)	2023-02-28 19:40:57 +01:00
rag	Better TF docstring types (#23477)	2023-05-24 13:52:52 +01:00
realm	🔥Rework pipeline testing by removing PipelineTestCaseMeta 🚀 (#21516)	2023-02-28 19:40:57 +01:00
reformer	Generate: skip left-padding tests on old models (#23437)	2023-05-18 11:04:51 +01:00
regnet	Better TF docstring types (#23477)	2023-05-24 13:52:52 +01:00
rembert	Better TF docstring types (#23477)	2023-05-24 13:52:52 +01:00
resnet	Better TF docstring types (#23477)	2023-05-24 13:52:52 +01:00
retribert	Update quality tooling for formatting (#21480)	2023-02-06 18:10:56 -05:00
roberta	Better TF docstring types (#23477)	2023-05-24 13:52:52 +01:00
roberta_prelayernorm	Better TF docstring types (#23477)	2023-05-24 13:52:52 +01:00
roc_bert	Move is_pipeline_test_to_skip to specific model test classes (#21999)	2023-03-14 10:03:02 +01:00
roformer	Better TF docstring types (#23477)	2023-05-24 13:52:52 +01:00
rwkv	Update tiny models and pipeline tests (#23446)	2023-05-18 17:29:04 +02:00
sam	Better TF docstring types (#23477)	2023-05-24 13:52:52 +01:00
segformer	Better TF docstring types (#23477)	2023-05-24 13:52:52 +01:00
sew	🔥Rework pipeline testing by removing PipelineTestCaseMeta 🚀 (#21516)	2023-02-28 19:40:57 +01:00
sew_d	🔥Rework pipeline testing by removing PipelineTestCaseMeta 🚀 (#21516)	2023-02-28 19:40:57 +01:00
speech_encoder_decoder	Update quality tooling for formatting (#21480)	2023-02-06 18:10:56 -05:00
speech_to_text	Better TF docstring types (#23477)	2023-05-24 13:52:52 +01:00
speech_to_text_2	🔥Rework pipeline testing by removing PipelineTestCaseMeta 🚀 (#21516)	2023-02-28 19:40:57 +01:00
speecht5	is_batched fix for remaining 2-D numpy arrays (#23309)	2023-05-23 14:37:35 -04:00
splinter	Make tiny model creation + pipeline testing more robust (#22500)	2023-04-06 17:45:55 +02:00
squeezebert	🔥Rework pipeline testing by removing PipelineTestCaseMeta 🚀 (#21516)	2023-02-28 19:40:57 +01:00
swiftformer	Update tiny models and pipeline tests (#23446)	2023-05-18 17:29:04 +02:00
swin	Better TF docstring types (#23477)	2023-05-24 13:52:52 +01:00
swin2sr	Skip test_multi_gpu_data_parallel_forward for some model tests (#21991)	2023-03-07 14:23:36 +01:00
swinv2	Skip test_multi_gpu_data_parallel_forward for some model tests (#21991)	2023-03-07 14:23:36 +01:00
switch_transformers	update_pip_test_mapping (#22606)	2023-04-06 17:56:06 +02:00
t5	Better TF docstring types (#23477)	2023-05-24 13:52:52 +01:00
table_transformer	🔥Rework pipeline testing by removing PipelineTestCaseMeta 🚀 (#21516)	2023-02-28 19:40:57 +01:00
tapas	Better TF docstring types (#23477)	2023-05-24 13:52:52 +01:00
tapex	Update quality tooling for formatting (#21480)	2023-02-06 18:10:56 -05:00
time_series_transformer	[Time-Series] Autoformer model (#21891)	2023-05-30 10:23:32 +02:00
timesformer	🔥Rework pipeline testing by removing PipelineTestCaseMeta 🚀 (#21516)	2023-02-28 19:40:57 +01:00
trajectory_transformer	Time to Say Goodbye, torch 1.7 and 1.8 (#22291)	2023-03-21 19:22:01 +01:00
transfo_xl	Better TF docstring types (#23477)	2023-05-24 13:52:52 +01:00
trocr	Generate: skip left-padding tests on old models (#23437)	2023-05-18 11:04:51 +01:00
tvlt	is_batched fix for remaining 2-D numpy arrays (#23309)	2023-05-23 14:37:35 -04:00
unispeech	🔥Rework pipeline testing by removing PipelineTestCaseMeta 🚀 (#21516)	2023-02-28 19:40:57 +01:00
unispeech_sat	🔥Rework pipeline testing by removing PipelineTestCaseMeta 🚀 (#21516)	2023-02-28 19:40:57 +01:00
upernet	🔥Rework pipeline testing by removing PipelineTestCaseMeta 🚀 (#21516)	2023-02-28 19:40:57 +01:00
van	🔥Rework pipeline testing by removing PipelineTestCaseMeta 🚀 (#21516)	2023-02-28 19:40:57 +01:00
videomae	🔥Rework pipeline testing by removing PipelineTestCaseMeta 🚀 (#21516)	2023-02-28 19:40:57 +01:00
vilt	🔥Rework pipeline testing by removing PipelineTestCaseMeta 🚀 (#21516)	2023-02-28 19:40:57 +01:00
vision_encoder_decoder	Better TF docstring types (#23477)	2023-05-24 13:52:52 +01:00
vision_text_dual_encoder	Better TF docstring types (#23477)	2023-05-24 13:52:52 +01:00
visual_bert	🔥Rework pipeline testing by removing PipelineTestCaseMeta 🚀 (#21516)	2023-02-28 19:40:57 +01:00
vit	Better TF docstring types (#23477)	2023-05-24 13:52:52 +01:00
vit_hybrid	🔥Rework pipeline testing by removing PipelineTestCaseMeta 🚀 (#21516)	2023-02-28 19:40:57 +01:00
vit_mae	Better TF docstring types (#23477)	2023-05-24 13:52:52 +01:00
vit_msn	🔥Rework pipeline testing by removing PipelineTestCaseMeta 🚀 (#21516)	2023-02-28 19:40:57 +01:00
wav2vec2	Better TF docstring types (#23477)	2023-05-24 13:52:52 +01:00
wav2vec2_conformer	🔥Rework pipeline testing by removing PipelineTestCaseMeta 🚀 (#21516)	2023-02-28 19:40:57 +01:00
wav2vec2_phoneme	Move test model folders (#17034)	2022-05-03 14:42:02 +02:00
wav2vec2_with_lm	Fix test_word_time_stamp_integration for Wav2Vec2ProcessorWithLMTest (#22800)	2023-04-17 12:41:55 +02:00
wavlm	🔥Rework pipeline testing by removing PipelineTestCaseMeta 🚀 (#21516)	2023-02-28 19:40:57 +01:00
whisper	fix Whisper tests on GPU (#23753)	2023-05-30 09:06:58 -04:00
x_clip	🔥Rework pipeline testing by removing PipelineTestCaseMeta 🚀 (#21516)	2023-02-28 19:40:57 +01:00
xglm	Better TF docstring types (#23477)	2023-05-24 13:52:52 +01:00
xlm	Better TF docstring types (#23477)	2023-05-24 13:52:52 +01:00
xlm_prophetnet	Update expected values in XLMProphetNetModelIntegrationTest (#21957)	2023-03-06 09:15:44 +01:00
xlm_roberta	Better TF docstring types (#23477)	2023-05-24 13:52:52 +01:00
xlm_roberta_xl	Use real tokenizers if tiny version(s) creation has issue(s) (#22428)	2023-03-29 16:16:23 +02:00
xlnet	Better TF docstring types (#23477)	2023-05-24 13:52:52 +01:00
xmod	Use real tokenizers if tiny version(s) creation has issue(s) (#22428)	2023-03-29 16:16:23 +02:00
yolos	[DETR, YOLOS] Fix device bug (#21974)	2023-03-07 07:34:04 -05:00
yoso	🔥Rework pipeline testing by removing PipelineTestCaseMeta 🚀 (#21516)	2023-02-28 19:40:57 +01:00
__init__.py	Move test model folders (#17034)	2022-05-03 14:42:02 +02:00