SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity
Files
8635407bc724c45142c1f91dbc9ef3ea681e1a56
HuggingFace_transformer/tests
History
Yih-Dar 8635407bc7 Fix tf.concatenate + test past_key_values for TF models (#15774) 
* fix wrong method name tf.concatenate

* add tests related to causal LM / decoder

* make style and quality

* clean-up

* Fix TFBertModel's extended_attention_mask when past_key_values is provided

* Fix tests

* fix copies

* More tf.int8 -> tf.int32 in TF test template

* clean-up

* Update TF test template

* revert the previous commit + update the TF test template

* Fix TF template extended_attention_mask when past_key_values is provided

* Fix some styles manually

* clean-up

* Fix ValueError: too many values to unpack in the test

* Fix more: too many values to unpack in the test

* Add a comment for extended_attention_mask when there is past_key_values

* Fix TFElectra extended_attention_mask when past_key_values is provided

* Add tests to other TF models

* Fix for TF Electra test: add prepare_config_and_inputs_for_decoder

* Fix not passing training arg to lm_head in TFRobertaForCausalLM

* Fix tests (with past) for TF Roberta

* add testing for pask_key_values for TFElectra model

Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
2022-02-25 17:11:46 +01:00
..
albert
[Test refactor 1/5] Per-folder tests reorganization (#15725)
2022-02-23 15:46:28 -05:00
auto
[Test refactor 1/5] Per-folder tests reorganization (#15725)
2022-02-23 15:46:28 -05:00
bart
[Test refactor 1/5] Per-folder tests reorganization (#15725)
2022-02-23 15:46:28 -05: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
[Test refactor 1/5] Per-folder tests reorganization (#15725)
2022-02-23 15:46:28 -05:00
blenderbot_small
[Test refactor 1/5] Per-folder tests reorganization (#15725)
2022-02-23 15:46:28 -05: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
[Test refactor 1/5] Per-folder tests reorganization (#15725)
2022-02-23 15:46:28 -05:00
convbert
[Test refactor 1/5] Per-folder tests reorganization (#15725)
2022-02-23 15:46:28 -05:00
convnext
[Test refactor 1/5] Per-folder tests reorganization (#15725)
2022-02-23 15:46:28 -05: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
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
[Test refactor 1/5] Per-folder tests reorganization (#15725)
2022-02-23 15:46:28 -05:00
deit
[Test refactor 1/5] Per-folder tests reorganization (#15725)
2022-02-23 15:46:28 -05:00
detr
[Test refactor 1/5] Per-folder tests reorganization (#15725)
2022-02-23 15:46:28 -05:00
distilbert
[Test refactor 1/5] Per-folder tests reorganization (#15725)
2022-02-23 15:46:28 -05: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
[Test refactor 1/5] Per-folder tests reorganization (#15725)
2022-02-23 15:46:28 -05: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
[Test refactor 1/5] Per-folder tests reorganization (#15725)
2022-02-23 15:46:28 -05:00
gpt2
[Test refactor 1/5] Per-folder tests reorganization (#15725)
2022-02-23 15:46:28 -05: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
[Test refactor 1/5] Per-folder tests reorganization (#15725)
2022-02-23 15:46:28 -05:00
layoutlmv2
[Test refactor 1/5] Per-folder tests reorganization (#15725)
2022-02-23 15:46:28 -05:00
layoutxlm
[Test refactor 1/5] Per-folder tests reorganization (#15725)
2022-02-23 15:46:28 -05:00
led
[Test refactor 1/5] Per-folder tests reorganization (#15725)
2022-02-23 15:46:28 -05: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
[Test refactor 1/5] Per-folder tests reorganization (#15725)
2022-02-23 15:46:28 -05: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
[Test refactor 1/5] Per-folder tests reorganization (#15725)
2022-02-23 15:46:28 -05: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
[Test refactor 1/5] Per-folder tests reorganization (#15725)
2022-02-23 15:46:28 -05: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
Adding the option to return_timestamps on pure CTC ASR models. (#15792)
2022-02-25 14:06:45 +01:00
plbart
[Test refactor 1/5] Per-folder tests reorganization (#15725)
2022-02-23 15:46:28 -05:00
poolformer
[Test refactor 1/5] Per-folder tests reorganization (#15725)
2022-02-23 15:46:28 -05: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
[Test refactor 1/5] Per-folder tests reorganization (#15725)
2022-02-23 15:46:28 -05: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
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
[Test refactor 1/5] Per-folder tests reorganization (#15725)
2022-02-23 15:46:28 -05:00
speech_to_text
[Test refactor 1/5] Per-folder tests reorganization (#15725)
2022-02-23 15:46:28 -05: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
[Test refactor 1/5] Per-folder tests reorganization (#15725)
2022-02-23 15:46:28 -05:00
t5
[Test refactor 1/5] Per-folder tests reorganization (#15725)
2022-02-23 15:46:28 -05: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
[Test refactor 1/5] Per-folder tests reorganization (#15725)
2022-02-23 15:46:28 -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
[Unispeech] Fix slow tests (#15818)
2022-02-24 19:08:54 +01:00
utils
[Test refactor 1/5] Per-folder tests reorganization (#15725)
2022-02-23 15:46:28 -05:00
vilt
[Test refactor 1/5] Per-folder tests reorganization (#15725)
2022-02-23 15:46:28 -05:00
vision_encoder_decoder
[Test refactor 1/5] Per-folder tests reorganization (#15725)
2022-02-23 15:46:28 -05: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
[Test refactor 1/5] Per-folder tests reorganization (#15725)
2022-02-23 15:46:28 -05:00
wav2vec2
[Test refactor 1/5] Per-folder tests reorganization (#15725)
2022-02-23 15:46:28 -05:00
wav2vec2_phoneme
[Test refactor 1/5] Per-folder tests reorganization (#15725)
2022-02-23 15:46:28 -05:00
wav2vec2_with_lm
[Test refactor 1/5] Per-folder tests reorganization (#15725)
2022-02-23 15:46:28 -05:00
wavlm
[Test refactor 1/5] Per-folder tests reorganization (#15725)
2022-02-23 15:46:28 -05:00
xglm
[Test refactor 1/5] Per-folder tests reorganization (#15725)
2022-02-23 15:46:28 -05: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
[Test refactor 1/5] Per-folder tests reorganization (#15725)
2022-02-23 15:46:28 -05: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
Fix _configuration_file argument getting passed to model (#15629)
2022-02-11 13:46:08 -05:00
test_feature_extraction_common.py
Allow custom code for Processors (#15649)
2022-02-15 09:44:35 -05:00
test_modeling_common.py
Fix from_pretrained with default base_model_prefix (#15814)
2022-02-24 11:43:51 +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
TF generate refactor - Greedy Search (#15562)
2022-02-15 17:54:43 +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