..
albert
Move test model folders ( #17034 )
2022-05-03 14:42:02 +02:00
auto
Throw an error if getattribute_from_module can't find anything ( #19535 )
2022-10-12 20:09:45 +02:00
bart
TF: TF 2.10 unpin + related onnx test skips ( #18995 )
2022-09-12 19:30:27 +01: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
Clean up deprecation warnings ( #19654 )
2022-10-18 13:34:47 -04:00
bert
Allow usage of TF Text BertTokenizer on TFBertTokenizer to make it servable on TF Serving ( #19590 )
2022-10-14 15:18:02 +01:00
bert_generation
Black preview ( #17217 )
2022-05-12 16:25:55 -04: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
Black preview ( #17217 )
2022-05-12 16:25:55 -04:00
blenderbot
🚨 🚨 🚨 TF: Remove TFWrappedEmbeddings (breaking: TF embedding initialization updated for encoder-decoder models) ( #19263 )
2022-10-11 16:48:03 +01:00
blenderbot_small
🚨 🚨 🚨 TF: Remove TFWrappedEmbeddings (breaking: TF embedding initialization updated for encoder-decoder models) ( #19263 )
2022-10-11 16:48:03 +01:00
bloom
Skip BloomEmbeddingTest.test_embeddings for PyTorch < 1.10 ( #19261 )
2022-10-10 10:05:30 +02: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
Replace as_target context managers by direct calls ( #18325 )
2022-07-29 08:09:09 -04:00
clip
Update serving code to enable saved_model=True ( #18153 )
2022-07-22 18:05:38 +01:00
codegen
Update expected values in CodeGen tests ( #17888 )
2022-07-01 15:33:36 +02:00
conditional_detr
Improve DETR models ( #19644 )
2022-10-18 10:29:14 +02:00
convbert
wrap forward passes with torch.no_grad() ( #19274 )
2022-10-04 16:12:03 +02:00
convnext
FX support for ConvNext, Wav2Vec2 and ResNet ( #19053 )
2022-09-16 10:57:41 +02:00
cpm
Move test model folders ( #17034 )
2022-05-03 14:42:02 +02:00
ctrl
Fix CTRL tests ( #17508 )
2022-06-01 16:27:23 +02:00
cvt
[CvT] Tensorflow implementation ( #18597 )
2022-10-11 18:16:52 +01:00
data2vec
skip some test_multi_gpu_data_parallel_forward ( #18188 )
2022-07-20 15:54:44 +02: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
wrap forward passes with torch.no_grad() ( #19278 )
2022-10-04 16:09:23 +02:00
decision_transformer
Update expected values in DecisionTransformerModelIntegrationTest ( #18016 )
2022-07-05 14:53:43 +02:00
deformable_detr
[fix] Add DeformableDetrFeatureExtractor ( #19140 )
2022-09-22 09:45:24 +02:00
deit
wrap forward passes with torch.no_grad() ( #19279 )
2022-10-04 16:08:29 +02:00
detr
[fix] Add DeformableDetrFeatureExtractor ( #19140 )
2022-09-22 09:45:24 +02: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 post_process_semantic_segmentation method to DPTFeatureExtractor ( #19107 )
2022-09-21 15:15:26 +03:00
electra
Move test model folders ( #17034 )
2022-05-03 14:42:02 +02:00
encoder_decoder
PT <-> TF for composite models ( #19732 )
2022-10-21 12:40:39 +02:00
ernie
add task_type_id to BERT to support ERNIE-2.0 and ERNIE-3.0 models ( #18686 )
2022-09-09 07:36:46 -04:00
esm
TF port of ESM ( #19587 )
2022-10-17 14:16:16 +01:00
flaubert
wrap forward passes with torch.no_grad() ( #19412 )
2022-10-10 15:04:10 -04:00
flava
Clean up deprecation warnings ( #19654 )
2022-10-18 13:34:47 -04:00
fnet
wrap forward passes with torch.no_grad() ( #19413 )
2022-10-10 15:03:46 -04:00
fsmt
Not use -1e4 as attn mask ( #17306 )
2022-06-20 16:16:16 +02: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
fix train_new_from_iterator in the case of byte-level tokenizers ( #17549 )
2022-06-08 15:30:41 +02: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
Fixing OPT fast tokenizer option. ( #18753 )
2022-09-15 17:12:58 +02:00
gptj
TF: GPT-J compatible with XLA generation ( #17986 )
2022-07-06 15:02:07 +01:00
groupvit
Fix TFGroupViT CI ( #19461 )
2022-10-11 14:29:15 +02: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
fix train_new_from_iterator in the case of byte-level tokenizers ( #17549 )
2022-06-08 15:30:41 +02:00
imagegpt
wrap forward passes with torch.no_grad() ( #19414 )
2022-10-10 15:03:24 -04:00
layoutlm
TF: TF 2.10 unpin + related onnx test skips ( #18995 )
2022-09-12 19:30:27 +01:00
layoutlmv2
add warning to let the user know that the __call__ method is faster than encode + pad for a fast tokenizer ( #18693 )
2022-08-24 06:27:56 -04:00
layoutlmv3
Fix test_tf_encode_plus_sent_to_model for LayoutLMv3 ( #18898 )
2022-09-06 14:51:03 +02:00
layoutxlm
LayoutXLMProcessor: ensure 1-to-1 mapping between samples and images, and add test for it ( #18774 )
2022-08-30 14:43:14 +02:00
led
Clean up deprecation warnings ( #19654 )
2022-10-18 13:34:47 -04:00
levit
Add skip logic for attentions test - Levit ( #17633 )
2022-06-10 12:46:30 +02:00
lilt
Add LiLT ( #19450 )
2022-10-12 10:11:20 +02:00
longformer
Fix tflongformer int dtype ( #18907 )
2022-09-12 17:51:10 +01:00
longt5
Skip test_export_to_onnx for LongT5 if torch < 1.11 ( #19122 )
2022-09-20 21:52:18 +02: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
Replace as_target context managers by direct calls ( #18325 )
2022-07-29 08:09:09 -04:00
marian
🚨 🚨 🚨 TF: Remove TFWrappedEmbeddings (breaking: TF embedding initialization updated for encoder-decoder models) ( #19263 )
2022-10-11 16:48:03 +01:00
markuplm
Add MarkupLM ( #19198 )
2022-09-30 08:25:43 +02:00
maskformer
Fix MaskFormer failing postprocess tests ( #19354 )
2022-10-05 23:25:58 +03:00
mbart
🚨 🚨 🚨 TF: Remove TFWrappedEmbeddings (breaking: TF embedding initialization updated for encoder-decoder models) ( #19263 )
2022-10-11 16:48:03 +01:00
mbart50
Replace as_target context managers by direct calls ( #18325 )
2022-07-29 08:09:09 -04:00
mctct
Replace as_target context managers by direct calls ( #18325 )
2022-07-29 08:09:09 -04: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
mobilevit
Add semantic segmentation post-processing method to MobileViT ( #19105 )
2022-09-23 16:24:28 +03: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
Replace as_target context managers by direct calls ( #18325 )
2022-07-29 08:09:09 -04:00
nezha
speed up test ( #18106 )
2022-07-12 04:28:28 -04: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
Move test model folders ( #17034 )
2022-05-03 14:42:02 +02:00
opt
Add OPTForQuestionAnswering ( #19402 )
2022-10-10 09:30:59 -04:00
owlvit
fix owlvit tests, update docstring examples ( #18586 )
2022-08-11 19:10:25 +03:00
pegasus
🚨 🚨 🚨 TF: Remove TFWrappedEmbeddings (breaking: TF embedding initialization updated for encoder-decoder models) ( #19263 )
2022-10-11 16:48:03 +01:00
pegasus_x
Fix CI for PegasusX ( #19025 )
2022-09-14 14:45:00 +02:00
perceiver
Replace as_target context managers by direct calls ( #18325 )
2022-07-29 08:09:09 -04:00
phobert
Move test model folders ( #17034 )
2022-05-03 14:42:02 +02:00
plbart
Replace as_target context managers by direct calls ( #18325 )
2022-07-29 08:09:09 -04:00
poolformer
has_attentions - consistent test skipping logic and tf tests ( #17495 )
2022-06-09 09:50:03 +02:00
prophetnet
Black preview ( #17217 )
2022-05-12 16:25:55 -04: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
Black preview ( #17217 )
2022-05-12 16:25:55 -04: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
fix train_new_from_iterator in the case of byte-level tokenizers ( #17549 )
2022-06-08 15:30:41 +02:00
roformer
wrap forward passes with torch.no_grad() ( #19438 )
2022-10-10 14:54:54 -04:00
segformer
Add post_process_semantic_segmentation method to SegFormer ( #19072 )
2022-09-21 11:40:35 +03: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
send model to the correct device ( #18800 )
2022-08-29 18:46:30 +02:00
speech_to_text
remvoe _create_and_check_torch_fx_tracing in specific test files ( #18667 )
2022-09-07 16:22:09 +02:00
speech_to_text_2
Fx support for multiple model architectures ( #17393 )
2022-05-31 10:02:55 +02: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
🚨 🚨 🚨 TF: Remove TFWrappedEmbeddings (breaking: TF embedding initialization updated for encoder-decoder models) ( #19263 )
2022-10-11 16:48:03 +01:00
table_transformer
Add table transformer [v2] ( #19614 )
2022-10-18 15:20:09 +02:00
tapas
Fix test_tf_encode_plus_sent_to_model for TAPAS ( #19559 )
2022-10-14 16:10:36 +02: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
Add trajectory transformer ( #17141 )
2022-05-17 19:07:43 -04:00
transfo_xl
Add magic method to our TF models to convert datasets with column inference ( #17160 )
2022-06-06 15:53:49 +01:00
trocr
Fx support for multiple model architectures ( #17393 )
2022-05-31 10:02:55 +02: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
Use repo_type instead of deprecated datasets repo IDs ( #19202 )
2022-09-26 09:50:48 -04:00
vilt
Add accelerate support for ViLT ( #18683 )
2022-09-22 13:14:39 +02:00
vision_encoder_decoder
PT <-> TF for composite models ( #19732 )
2022-10-21 12:40:39 +02:00
vision_text_dual_encoder
Move test model folders ( #17034 )
2022-05-03 14:42:02 +02:00
visual_bert
wrap forward passes with torch.no_grad() ( #19439 )
2022-10-10 14:54:36 -04:00
vit
Improve vision models ( #17731 )
2022-06-24 11:34:51 +02:00
vit_mae
TF: tests for (de)serializable models with resized tokens ( #19013 )
2022-09-16 16:38:08 +01:00
vit_msn
fix: ckpt paths. ( #19159 )
2022-09-22 11:03:01 -04:00
wav2vec2
Clean up deprecation warnings ( #19654 )
2022-10-18 13:34:47 -04: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
Allow user-managed Pool in Wav2Vec2ProcessorWithLM.batch_decode ( #18351 )
2022-10-18 08:48:03 -04:00
wavlm
Move test model folders ( #17034 )
2022-05-03 14:42:02 +02:00
whisper
fix tests ( #19670 )
2022-10-18 06:45:48 +02:00
x_clip
[X-CLIP] Fix doc tests ( #19523 )
2022-10-12 17:05:12 +02:00
xglm
XGLM - Fix Softmax NaNs when using FP16 ( #18057 )
2022-09-29 10:42:07 +02:00
xlm
Move test model folders ( #17034 )
2022-05-03 14:42:02 +02: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
Move test model folders ( #17034 )
2022-05-03 14:42:02 +02:00
xlnet
Return scalar losses instead of per-sample means ( #18013 )
2022-07-04 17:26:19 +01: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
31565ff0fd