transformers/tests
2022-04-07 11:05:10 -04:00
..
albert Use random_attention_mask for TF tests (#16517) 2022-04-01 16:53:07 +02:00
auto Remove parent/child tests in auto model tests (#16653) 2022-04-07 11:05:10 -04:00
bart Reorganize file utils (#16264) 2022-03-23 10:26:33 -04:00
barthez
bartpho
beit Reorganize file utils (#16264) 2022-03-23 10:26:33 -04:00
benchmark
bert Use random_attention_mask for TF tests (#16517) 2022-04-01 16:53:07 +02:00
bert_generation Reorganize file utils (#16264) 2022-03-23 10:26:33 -04:00
bert_japanese
bertweet
big_bird Fix missing output_attentions in PT/Flax equivalence test (#16271) 2022-03-29 17:51:48 +02:00
bigbird_pegasus
blenderbot Reorganize file utils (#16264) 2022-03-23 10:26:33 -04:00
blenderbot_small Reorganize file utils (#16264) 2022-03-23 10:26:33 -04:00
bort
byt5 add a test checking the format of convert_tokens_to_string's output (#16540) 2022-04-04 16:57:24 +02:00
camembert Reorganize file utils (#16264) 2022-03-23 10:26:33 -04:00
canine Reorganize file utils (#16264) 2022-03-23 10:26:33 -04:00
clip Use random_attention_mask for TF tests (#16517) 2022-04-01 16:53:07 +02:00
convbert TF: Finalize unpack_inputs-related changes (#16499) 2022-04-04 16:37:33 +01:00
convnext Adding new train_step logic to make things less confusing for users (#15994) 2022-04-05 14:23:27 +01:00
cpm
ctrl Use random_attention_mask for TF tests (#16517) 2022-04-01 16:53:07 +02:00
data2vec
deberta Use random_attention_mask for TF tests (#16517) 2022-04-01 16:53:07 +02:00
deberta_v2 Use random_attention_mask for TF tests (#16517) 2022-04-01 16:53:07 +02:00
decision_transformer Decision transformer gym (#15845) 2022-03-23 16:18:43 -04:00
deepspeed Reorganize file utils (#16264) 2022-03-23 10:26:33 -04:00
deit Reorganize file utils (#16264) 2022-03-23 10:26:33 -04:00
detr Reorganize file utils (#16264) 2022-03-23 10:26:33 -04:00
distilbert Use random_attention_mask for TF tests (#16517) 2022-04-01 16:53:07 +02:00
dit [Tests] Fix DiT test (#16218) 2022-03-17 10:53:57 +01:00
dpr Use random_attention_mask for TF tests (#16517) 2022-04-01 16:53:07 +02:00
dpt Add DPT (#15991) 2022-03-28 16:28:10 +02:00
electra Use random_attention_mask for TF tests (#16517) 2022-04-01 16:53:07 +02:00
encoder_decoder TF: properly handle kwargs in encoder_decoder architectures (#16465) 2022-03-29 18:17:47 +01:00
extended Reorganize file utils (#16264) 2022-03-23 10:26:33 -04:00
fixtures
flaubert Use random_attention_mask for TF tests (#16517) 2022-04-01 16:53:07 +02:00
fnet
fsmt Reorganize file utils (#16264) 2022-03-23 10:26:33 -04:00
funnel Use random_attention_mask for TF tests (#16517) 2022-04-01 16:53:07 +02:00
generation TF generate refactor - Beam Search (#16374) 2022-04-06 18:19:34 +01:00
glpn Reorganize file utils (#16264) 2022-03-23 10:26:33 -04:00
gpt2 TF generate refactor - Beam Search (#16374) 2022-04-06 18:19:34 +01:00
gpt_neo Reorganize file utils (#16264) 2022-03-23 10:26:33 -04:00
gptj Use random_attention_mask for TF tests (#16517) 2022-04-01 16:53:07 +02:00
herbert
hubert
ibert
imagegpt Reorganize file utils (#16264) 2022-03-23 10:26:33 -04:00
layoutlm Use random_attention_mask for TF tests (#16517) 2022-04-01 16:53:07 +02:00
layoutlmv2 Reorganize file utils (#16264) 2022-03-23 10:26:33 -04:00
layoutxlm Reorganize file utils (#16264) 2022-03-23 10:26:33 -04:00
led Reorganize file utils (#16264) 2022-03-23 10:26:33 -04:00
longformer Use random_attention_mask for TF tests (#16517) 2022-04-01 16:53:07 +02:00
luke
lxmert Use random_attention_mask for TF tests (#16517) 2022-04-01 16:53:07 +02:00
m2m_100 Reorganize file utils (#16264) 2022-03-23 10:26:33 -04:00
marian Reorganize file utils (#16264) 2022-03-23 10:26:33 -04:00
maskformer Feature Extractor accepts segmentation_maps (#15964) 2022-03-30 18:46:51 +02:00
mbart Reorganize file utils (#16264) 2022-03-23 10:26:33 -04:00
mbart50
megatron_bert
megatron_gpt2
mluke
mobilebert Use random_attention_mask for TF tests (#16517) 2022-04-01 16:53:07 +02:00
mpnet Use random_attention_mask for TF tests (#16517) 2022-04-01 16:53:07 +02:00
mt5
nystromformer
onnx Add ONNX export for BeiT (#16498) 2022-04-01 10:52:42 +02:00
openai Use random_attention_mask for TF tests (#16517) 2022-04-01 16:53:07 +02:00
optimization
pegasus Reorganize file utils (#16264) 2022-03-23 10:26:33 -04:00
perceiver add a test checking the format of convert_tokens_to_string's output (#16540) 2022-04-04 16:57:24 +02:00
phobert
pipelines Attention mask is important in the case of batching... (#16222) 2022-03-18 10:02:12 +01:00
plbart Reorganize file utils (#16264) 2022-03-23 10:26:33 -04:00
poolformer Reorganize file utils (#16264) 2022-03-23 10:26:33 -04:00
prophetnet
qdqbert
rag Reorganize file utils (#16264) 2022-03-23 10:26:33 -04:00
realm
reformer Reorganize file utils (#16264) 2022-03-23 10:26:33 -04:00
rembert Use random_attention_mask for TF tests (#16517) 2022-04-01 16:53:07 +02:00
resnet Reorganize file utils (#16264) 2022-03-23 10:26:33 -04:00
roberta Use random_attention_mask for TF tests (#16517) 2022-04-01 16:53:07 +02:00
roformer Use random_attention_mask for TF tests (#16517) 2022-04-01 16:53:07 +02:00
sagemaker Adapt import to new structure 2022-03-24 14:40:05 -04:00
segformer Reorganize file utils (#16264) 2022-03-23 10:26:33 -04:00
sew
sew_d
speech_encoder_decoder [FlaxSpeechEncoderDecoderModel] More Rigorous PT-Flax Equivalence Tests (#16589) 2022-04-06 15:33:32 +02:00
speech_to_text TF generate refactor - Beam Search (#16374) 2022-04-06 18:19:34 +01:00
speech_to_text_2
splinter
squeezebert
swin Reorganize file utils (#16264) 2022-03-23 10:26:33 -04:00
t5 Adding new train_step logic to make things less confusing for users (#15994) 2022-04-05 14:23:27 +01:00
tapas Use random_attention_mask for TF tests (#16517) 2022-04-01 16:53:07 +02:00
tokenization
trainer Avoid accessing .dataset of a DataLoader in Trainer (#16451) 2022-03-29 15:00:18 -04:00
transfo_xl Fixes Loss for TransfoXL when using Trainer API v2 (#16140) 2022-03-17 05:49:24 -04:00
trocr
unispeech
unispeech_sat
utils Add utility to find model labels (#16526) 2022-04-04 10:06:57 -04:00
van Reorganize file utils (#16264) 2022-03-23 10:26:33 -04:00
vilt Reorganize file utils (#16264) 2022-03-23 10:26:33 -04:00
vision_encoder_decoder TF: properly handle kwargs in encoder_decoder architectures (#16465) 2022-03-29 18:17:47 +01:00
vision_text_dual_encoder Reorganize file utils (#16264) 2022-03-23 10:26:33 -04:00
visual_bert
vit Reorganize file utils (#16264) 2022-03-23 10:26:33 -04:00
vit_mae Fix CI: test_inference_for_pretraining in ViTMAEModelTest (#16591) 2022-04-05 10:00:03 +02:00
wav2vec2 add a test checking the format of convert_tokens_to_string's output (#16540) 2022-04-04 16:57:24 +02:00
wav2vec2_phoneme add a test checking the format of convert_tokens_to_string's output (#16540) 2022-04-04 16:57:24 +02:00
wav2vec2_with_lm Reorganize file utils (#16264) 2022-03-23 10:26:33 -04:00
wavlm
xglm Reorganize file utils (#16264) 2022-03-23 10:26:33 -04:00
xlm Use random_attention_mask for TF tests (#16517) 2022-04-01 16:53:07 +02:00
xlm_prophetnet Reorganize file utils (#16264) 2022-03-23 10:26:33 -04:00
xlm_roberta Reorganize file utils (#16264) 2022-03-23 10:26:33 -04:00
xlm_roberta_xl
xlnet Use random_attention_mask for TF tests (#16517) 2022-04-01 16:53:07 +02:00
yoso
__init__.py
test_configuration_common.py Make Transformers use cache files when hf.co is down (#16362) 2022-03-23 15:56:49 -04:00
test_feature_extraction_common.py Make Transformers use cache files when hf.co is down (#16362) 2022-03-23 15:56:49 -04:00
test_modeling_common.py Add DPT (#15991) 2022-03-28 16:28:10 +02:00
test_modeling_flax_common.py Fix missing output_attentions in PT/Flax equivalence test (#16271) 2022-03-29 17:51:48 +02:00
test_modeling_tf_common.py TF generate refactor - Beam Search (#16374) 2022-04-06 18:19:34 +01:00
test_sequence_feature_extraction_common.py
test_tokenization_common.py add a test checking the format of convert_tokens_to_string's output (#16540) 2022-04-04 16:57:24 +02:00