SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity
Files
667b823b894700ec848c4e0bdcfc65630157ef82
HuggingFace_transformer/tests
History
Francesco Saverio Zuppichini 667b823b89 Swin support for any input size (#15986) 
* padding done

* correctly return one attention per layer

* almost correct, attentions are not flatten one tuple per stage

* tests green

* doc

* conversations

* reshaping hidden_states

* view in the test

* reshape_hidden_states in Encoder and Model

* new outputs with reshaped_hidden_states

* conversations

* doc

* Update docs/source/model_doc/swin.mdx

Co-authored-by: NielsRogge <48327001+NielsRogge@users.noreply.github.com>

* Apply suggestions from code review

Co-authored-by: NielsRogge <48327001+NielsRogge@users.noreply.github.com>

* conversations

* fix tests

* minor changes

* resolved conversations

* attentions one per stage

* typo

* typos

* typos

* function signature

* CI

* clean up tests

Co-authored-by: NielsRogge <48327001+NielsRogge@users.noreply.github.com>
2022-03-16 18:38:25 +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
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
Replace all deprecated jax.ops operations with jnp's at (#16078)
2022-03-16 09:08:55 +00: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