transformers/tests
Yih-Dar f466936476
Add has_attentions to TFModelTesterMixin as done on PyTorch side (#16259)
Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
2022-03-19 11:44:17 +01:00
..
albert [Test refactor 1/5] Per-folder tests reorganization (#15725) 2022-02-23 15:46:28 -05:00
auto Removed an outdated check about hdf5_version (#16011) 2022-03-09 14:21:23 +01:00
bart Add FlaxBartForCausalLM (#15995) 2022-03-09 19:53:01 +01:00
barthez [Test refactor 1/5] Per-folder tests reorganization (#15725) 2022-02-23 15:46:28 -05:00
bartpho [Test refactor 1/5] Per-folder tests reorganization (#15725) 2022-02-23 15:46:28 -05:00
beit Update expected slices for pillow > 9 (#16117) 2022-03-18 09:46:45 +01:00
benchmark [Test refactor 1/5] Per-folder tests reorganization (#15725) 2022-02-23 15:46:28 -05:00
bert Fix tf.concatenate + test past_key_values for TF models (#15774) 2022-02-25 17:11:46 +01:00
bert_generation [Test refactor 1/5] Per-folder tests reorganization (#15725) 2022-02-23 15:46:28 -05:00
bert_japanese [Test refactor 1/5] Per-folder tests reorganization (#15725) 2022-02-23 15:46:28 -05:00
bertweet [Test refactor 1/5] Per-folder tests reorganization (#15725) 2022-02-23 15:46:28 -05:00
big_bird [Test refactor 1/5] Per-folder tests reorganization (#15725) 2022-02-23 15:46:28 -05:00
bigbird_pegasus [Test refactor 1/5] Per-folder tests reorganization (#15725) 2022-02-23 15:46:28 -05:00
blenderbot TF generate refactor - past without encoder outputs (#15944) 2022-03-08 14:46:44 +00:00
blenderbot_small TF generate refactor - past without encoder outputs (#15944) 2022-03-08 14:46:44 +00:00
bort [Test refactor 1/5] Per-folder tests reorganization (#15725) 2022-02-23 15:46:28 -05:00
byt5 [Test refactor 1/5] Per-folder tests reorganization (#15725) 2022-02-23 15:46:28 -05:00
camembert [Test refactor 1/5] Per-folder tests reorganization (#15725) 2022-02-23 15:46:28 -05:00
canine [Test refactor 1/5] Per-folder tests reorganization (#15725) 2022-02-23 15:46:28 -05:00
clip Fix loading CLIPVisionConfig and CLIPTextConfig (#16198) 2022-03-16 16:24:01 +01:00
convbert [Test refactor 1/5] Per-folder tests reorganization (#15725) 2022-02-23 15:46:28 -05:00
convnext [Tests] Add attentions_option to ModelTesterMixin (#15909) 2022-03-10 12:00:30 +01:00
cpm [Test refactor 1/5] Per-folder tests reorganization (#15725) 2022-02-23 15:46:28 -05:00
ctrl [Test refactor 1/5] Per-folder tests reorganization (#15725) 2022-02-23 15:46:28 -05:00
data2vec Add Data2Vec (#15507) 2022-03-01 11:09:20 +01:00
deberta [Test refactor 1/5] Per-folder tests reorganization (#15725) 2022-02-23 15:46:28 -05:00
deberta_v2 [Test refactor 1/5] Per-folder tests reorganization (#15725) 2022-02-23 15:46:28 -05:00
deepspeed [Deepspeed] add support for bf16 mode (#14569) 2022-03-11 17:53:53 -08:00
deit [vision] Add problem_type support (#15851) 2022-03-01 18:09:52 +01:00
detr Mark slow tests as slow 2022-03-03 11:03:24 -05:00
distilbert [Test refactor 1/5] Per-folder tests reorganization (#15725) 2022-02-23 15:46:28 -05:00
dit [Tests] Fix DiT test (#16218) 2022-03-17 10:53:57 +01:00
dpr [Test refactor 1/5] Per-folder tests reorganization (#15725) 2022-02-23 15:46:28 -05:00
electra Fix tf.concatenate + test past_key_values for TF models (#15774) 2022-02-25 17:11:46 +01:00
encoder_decoder Fix Loading of Flax(Speech)EncoderDecoderModel kwargs from PreTrained Encoder-Decoder Checkpoints (#16056) 2022-03-14 10:12:29 +01:00
extended Fix failing GPU trainer tests (#14903) 2021-12-23 13:59:33 -05:00
fixtures Add model like (#14992) 2022-01-24 15:25:10 -05:00
flaubert [Test refactor 1/5] Per-folder tests reorganization (#15725) 2022-02-23 15:46:28 -05:00
fnet [Test refactor 1/5] Per-folder tests reorganization (#15725) 2022-02-23 15:46:28 -05:00
fsmt [Test refactor 1/5] Per-folder tests reorganization (#15725) 2022-02-23 15:46:28 -05:00
funnel [Test refactor 1/5] Per-folder tests reorganization (#15725) 2022-02-23 15:46:28 -05:00
generation Fix generation min length (#16206) 2022-03-16 18:49:23 +01:00
gpt_neo [Test refactor 1/5] Per-folder tests reorganization (#15725) 2022-02-23 15:46:28 -05:00
gpt2 TF: add beam search tests (#16202) 2022-03-16 15:44:33 +00:00
gptj [Test refactor 1/5] Per-folder tests reorganization (#15725) 2022-02-23 15:46:28 -05:00
herbert [Test refactor 1/5] Per-folder tests reorganization (#15725) 2022-02-23 15:46:28 -05:00
hubert [Test refactor 1/5] Per-folder tests reorganization (#15725) 2022-02-23 15:46:28 -05:00
ibert [Test refactor 1/5] Per-folder tests reorganization (#15725) 2022-02-23 15:46:28 -05:00
imagegpt [Test refactor 1/5] Per-folder tests reorganization (#15725) 2022-02-23 15:46:28 -05:00
layoutlm Fix a TF test name (LayoutLMModelTest) (#16061) 2022-03-11 11:22:36 +01:00
layoutlmv2 Fix LayoutLMv2 test (#15939) 2022-03-08 10:49:30 +01:00
layoutxlm [Test refactor 1/5] Per-folder tests reorganization (#15725) 2022-02-23 15:46:28 -05:00
led TF generate refactor - past without encoder outputs (#15944) 2022-03-08 14:46:44 +00:00
longformer [Test refactor 1/5] Per-folder tests reorganization (#15725) 2022-02-23 15:46:28 -05:00
luke [Test refactor 1/5] Per-folder tests reorganization (#15725) 2022-02-23 15:46:28 -05:00
lxmert [Test refactor 1/5] Per-folder tests reorganization (#15725) 2022-02-23 15:46:28 -05:00
m2m_100 [Test refactor 1/5] Per-folder tests reorganization (#15725) 2022-02-23 15:46:28 -05:00
marian support new marian models (#15831) 2022-03-10 19:41:56 +01:00
maskformer update test (#16219) 2022-03-17 08:11:55 -04:00
mbart [Test refactor 1/5] Per-folder tests reorganization (#15725) 2022-02-23 15:46:28 -05:00
mbart50 [Test refactor 1/5] Per-folder tests reorganization (#15725) 2022-02-23 15:46:28 -05:00
megatron_bert [Test refactor 1/5] Per-folder tests reorganization (#15725) 2022-02-23 15:46:28 -05:00
megatron_gpt2 [Test refactor 1/5] Per-folder tests reorganization (#15725) 2022-02-23 15:46:28 -05:00
mluke [Test refactor 1/5] Per-folder tests reorganization (#15725) 2022-02-23 15:46:28 -05:00
mobilebert [Test refactor 1/5] Per-folder tests reorganization (#15725) 2022-02-23 15:46:28 -05:00
mpnet [Test refactor 1/5] Per-folder tests reorganization (#15725) 2022-02-23 15:46:28 -05:00
mt5 [Test refactor 1/5] Per-folder tests reorganization (#15725) 2022-02-23 15:46:28 -05:00
nystromformer [Test refactor 1/5] Per-folder tests reorganization (#15725) 2022-02-23 15:46:28 -05:00
onnx Add TFCamembertForCausalLM and ONNX integration test (#16073) 2022-03-14 08:40:42 +01:00
openai [Test refactor 1/5] Per-folder tests reorganization (#15725) 2022-02-23 15:46:28 -05:00
optimization [Test refactor 1/5] Per-folder tests reorganization (#15725) 2022-02-23 15:46:28 -05:00
pegasus TF generate refactor - past without encoder outputs (#15944) 2022-03-08 14:46:44 +00:00
perceiver [Test refactor 1/5] Per-folder tests reorganization (#15725) 2022-02-23 15:46:28 -05:00
phobert [Test refactor 1/5] Per-folder tests reorganization (#15725) 2022-02-23 15:46:28 -05:00
pipelines Attention mask is important in the case of batching... (#16222) 2022-03-18 10:02:12 +01:00
plbart [Test refactor 1/5] Per-folder tests reorganization (#15725) 2022-02-23 15:46:28 -05:00
poolformer [Tests] Add attentions_option to ModelTesterMixin (#15909) 2022-03-10 12:00:30 +01:00
prophetnet [Test refactor 1/5] Per-folder tests reorganization (#15725) 2022-02-23 15:46:28 -05:00
qdqbert [Test refactor 1/5] Per-folder tests reorganization (#15725) 2022-02-23 15:46:28 -05:00
rag [Test refactor 1/5] Per-folder tests reorganization (#15725) 2022-02-23 15:46:28 -05:00
realm Fix and improve REALM fine-tuning (#15297) 2022-03-03 14:10:15 +01:00
reformer [Test refactor 1/5] Per-folder tests reorganization (#15725) 2022-02-23 15:46:28 -05:00
rembert Fix tf.concatenate + test past_key_values for TF models (#15774) 2022-02-25 17:11:46 +01:00
resnet [WIP] Resnet (#15770) 2022-03-14 19:57:55 +01:00
roberta Fix tf.concatenate + test past_key_values for TF models (#15774) 2022-02-25 17:11:46 +01:00
roformer [Test refactor 1/5] Per-folder tests reorganization (#15725) 2022-02-23 15:46:28 -05:00
sagemaker SageMaker: Fix sagemaker DDP & metric logs (#13181) 2021-08-23 10:18:07 +02:00
segformer Revert changes in logit size for semantic segmentation models (#15722) 2022-02-24 15:52:52 +01:00
sew [Test refactor 1/5] Per-folder tests reorganization (#15725) 2022-02-23 15:46:28 -05:00
sew_d [Test refactor 1/5] Per-folder tests reorganization (#15725) 2022-02-23 15:46:28 -05:00
speech_encoder_decoder update jax version and re-enable some tests (#16254) 2022-03-18 16:45:39 +01:00
speech_to_text TF generate refactor - past without encoder outputs (#15944) 2022-03-08 14:46:44 +00:00
speech_to_text_2 [Test refactor 1/5] Per-folder tests reorganization (#15725) 2022-02-23 15:46:28 -05:00
splinter [Test refactor 1/5] Per-folder tests reorganization (#15725) 2022-02-23 15:46:28 -05:00
squeezebert [Test refactor 1/5] Per-folder tests reorganization (#15725) 2022-02-23 15:46:28 -05:00
swin Swin support for any input size (#15986) 2022-03-16 18:38:25 +01:00
t5 TF: add beam search tests (#16202) 2022-03-16 15:44:33 +00:00
tapas [Test refactor 1/5] Per-folder tests reorganization (#15725) 2022-02-23 15:46:28 -05:00
tokenization [Test refactor 1/5] Per-folder tests reorganization (#15725) 2022-02-23 15:46:28 -05:00
trainer Seed _get_train_sampler's generator with arg seed to improve reproducibility (#15961) 2022-03-08 13:45:41 -05:00
transfo_xl Fixes Loss for TransfoXL when using Trainer API v2 (#16140) 2022-03-17 05:49:24 -04:00
trocr [Test refactor 1/5] Per-folder tests reorganization (#15725) 2022-02-23 15:46:28 -05:00
unispeech [Unispeech] Fix slow tests (#15818) 2022-02-24 19:08:54 +01:00
unispeech_sat [UniSpeechSat] correct unispeech sat (#15847) 2022-02-28 11:23:13 +01:00
utils Support PEP 563 for HfArgumentParser (#15795) 2022-03-17 13:51:37 -04:00
van Visual Attention Network (VAN) (#16027) 2022-03-15 08:47:12 +01:00
vilt Update expected slices for pillow > 9 (#16117) 2022-03-18 09:46:45 +01:00
vision_encoder_decoder Update expected slices for pillow > 9 (#16117) 2022-03-18 09:46:45 +01:00
vision_text_dual_encoder [Test refactor 1/5] Per-folder tests reorganization (#15725) 2022-02-23 15:46:28 -05:00
visual_bert [Test refactor 1/5] Per-folder tests reorganization (#15725) 2022-02-23 15:46:28 -05:00
vit [Test refactor 1/5] Per-folder tests reorganization (#15725) 2022-02-23 15:46:28 -05:00
vit_mae [Tests] Fix ViTMAE integration test (#15949) 2022-03-08 10:49:44 +01:00
wav2vec2 Backprop Test for Freeze FlaxWav2Vec2 Feature Encoder (#15938) 2022-03-07 18:10:15 +01:00
wav2vec2_phoneme [Test refactor 1/5] Per-folder tests reorganization (#15725) 2022-02-23 15:46:28 -05:00
wav2vec2_with_lm Add time stamps for wav2vec2 with lm (#15854) 2022-03-01 17:03:05 +01:00
wavlm [Test refactor 1/5] Per-folder tests reorganization (#15725) 2022-02-23 15:46:28 -05:00
xglm [XGLM] run sampling test on CPU to be deterministic (#15892) 2022-03-02 17:55:49 +01:00
xlm [Test refactor 1/5] Per-folder tests reorganization (#15725) 2022-02-23 15:46:28 -05:00
xlm_prophetnet [Test refactor 1/5] Per-folder tests reorganization (#15725) 2022-02-23 15:46:28 -05:00
xlm_roberta Add missing support for Flax XLM-RoBERTa (#15900) 2022-03-04 14:36:28 +01:00
xlm_roberta_xl [Test refactor 1/5] Per-folder tests reorganization (#15725) 2022-02-23 15:46:28 -05:00
xlnet [TFXLNet] Correct tf xlnet generate (#15822) 2022-02-24 19:23:34 +01:00
yoso [Test refactor 1/5] Per-folder tests reorganization (#15725) 2022-02-23 15:46:28 -05:00
__init__.py GPU text generation: mMoved the encoded_prompt to correct device 2020-01-06 15:11:12 +01:00
test_configuration_common.py Add soft length regulation for sequence generation (#15245) 2022-03-11 19:36:44 +01:00
test_feature_extraction_common.py Allow custom code for Processors (#15649) 2022-02-15 09:44:35 -05:00
test_modeling_common.py Aggressive PT/TF equivalence test on PT side (#16250) 2022-03-18 18:51:24 +01:00
test_modeling_flax_common.py Make Flax pt-flax equivalence test more aggressive (#15841) 2022-03-18 18:15:36 +01:00
test_modeling_tf_common.py Add has_attentions to TFModelTesterMixin as done on PyTorch side (#16259) 2022-03-19 11:44:17 +01:00
test_sequence_feature_extraction_common.py [SequenceFeatureExtractor] Rewrite padding logic from pure python to numpy (#13650) 2021-09-21 17:10:13 +03:00
test_tokenization_common.py Allow custom code for Processors (#15649) 2022-02-15 09:44:35 -05:00