HuggingFace_transformer

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

Revert changes in logit size for semantic segmentation models (#15722 )

2022-02-24 15:52:52 +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

Add Document Image Transformer (DiT) (#15984 )

2022-03-10 11:34:44 +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

gpt2

TF: add beam search tests (#16202 )

2022-03-16 15:44:33 +00:00

gpt_neo

[Test refactor 1/5] Per-folder tests reorganization (#15725 )

2022-02-23 15:46:28 -05: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

done (#16012 )

2022-03-09 15:51:56 +01: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

Add ForInstanceSegmentation models to image-segmentation pipelines (#15937 )

2022-03-09 10:19:05 +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

Fix Loading of Flax(Speech)EncoderDecoderModel kwargs from PreTrained Encoder-Decoder Checkpoints (#16056 )

2022-03-14 10:12:29 +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

[Test refactor 1/5] Per-folder tests reorganization (#15725 )

2022-02-23 15:46:28 -05: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

[Test refactor 1/5] Per-folder tests reorganization (#15725 )

2022-02-23 15:46:28 -05:00

van

Visual Attention Network (VAN) (#16027 )

2022-03-15 08:47:12 +01:00

vilt

[Test refactor 1/5] Per-folder tests reorganization (#15725 )

2022-02-23 15:46:28 -05:00

vision_encoder_decoder

Improve Swin for VisionEncoderDecoder (#16070 )

2022-03-15 09:59:48 +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

[Tests] Add attentions_option to ModelTesterMixin (#15909 )

2022-03-10 12:00:30 +01:00

test_modeling_flax_common.py

[Flax tests] fix test_model_outputs_equivalence (#15571 )

2022-02-09 12:26:48 +01:00

test_modeling_tf_common.py

Make TF pt-tf equivalence test more aggressive (#15839 )

2022-03-14 13:31:32 +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