SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity
Files
1ac698744c4dbdf1495d303246d08ffacdf4f5b8
HuggingFace_transformer/tests
History
NielsRogge 1ac698744c Add YOLOS (#16848) 
* First draft

* Add YolosForObjectDetection

* Make forward pass work

* Add mid position embeddings

* Add interpolation of position encodings

* Add expected values

* Add YOLOS to tests

* Add integration test

* Support tiny model as well

* Support all models in conversion script

* Remove mid_pe_size attribute

* Make more tests pass

* Add model to README and fix config

* Add copied from statements

* Rename base_model_prefix to vit

* Add missing YOLOS_PRETRAINED_CONFIG_ARCHIVE_MAP

* Apply suggestions from code review

* Apply more suggestions from code review

* Convert remaining checkpoints

* Improve docstrings

* Add YolosFeatureExtractor

* Add feature extractor to docs

* Add corresponding tests

* Fix style

* Fix docs

* Apply suggestion from code review

* Fix bad rebase

* Fix some more bad rebase

* Fix missing character

* Improve docs and variable names

Co-authored-by: Niels Rogge <nielsrogge@Nielss-MacBook-Pro.local>
2022-05-02 18:30:55 +02: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
[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
Clean up vision tests (#17024)
2022-05-02 16:28:58 +02:00
benchmark
[Test refactor 1/5] Per-folder tests reorganization (#15725)
2022-02-23 15:46:28 -05:00
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
Fix issue avoid-missing-comma found at https://codereview.doctor (#16768)
2022-04-14 16:42:27 -04:00
bertweet
[Test refactor 1/5] Per-folder tests reorganization (#15725)
2022-02-23 15:46:28 -05:00
big_bird
Fix missing output_attentions in PT/Flax equivalence test (#16271)
2022-03-29 17:51:48 +02:00
bigbird_pegasus
[Test refactor 1/5] Per-folder tests reorganization (#15725)
2022-02-23 15:46:28 -05:00
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
[Test refactor 1/5] Per-folder tests reorganization (#15725)
2022-02-23 15:46:28 -05:00
byt5
Fix #16660 (tokenizers setters of ids of special tokens) (#16661)
2022-04-13 07:49:06 -04:00
camembert
Reorganize file utils (#16264)
2022-03-23 10:26:33 -04:00
canine
Fix #16660 (tokenizers setters of ids of special tokens) (#16661)
2022-04-13 07:49:06 -04:00
clip
Improve test_pt_tf_model_equivalence on PT side (#16731)
2022-04-19 21:13:27 +02:00
convbert
TF: Finalize unpack_inputs-related changes (#16499)
2022-04-04 16:37:33 +01:00
convnext
Enable more test_torchscript (#16679)
2022-04-11 18:23:35 +02:00
cpm
[Test refactor 1/5] Per-folder tests reorganization (#15725)
2022-02-23 15:46:28 -05:00
ctrl
Enable more test_torchscript (#16679)
2022-04-11 18:23:35 +02:00
data2vec
use scale=1.0 in floats_tensor called in speech model testers (#17007)
2022-04-29 14:41:33 +02:00
deberta
Use random_attention_mask for TF tests (#16517)
2022-04-01 16:53:07 +02:00
deberta_v2
add DebertaV2 fast tokenizer (#15529)
2022-04-20 10:26:51 +02:00
decision_transformer
Enable more test_torchscript (#16679)
2022-04-11 18:23:35 +02:00
deepspeed
[trainer / deepspeed] fix hyperparameter_search (#16740)
2022-04-14 17:24:38 -07:00
deit
Clean up vision tests (#17024)
2022-05-02 16:28:58 +02:00
detr
Reorganize file utils (#16264)
2022-03-23 10:26:33 -04:00
distilbert
Enable more test_torchscript (#16679)
2022-04-11 18:23:35 +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
Clean up vision tests (#17024)
2022-05-02 16:28:58 +02:00
electra
Use random_attention_mask for TF tests (#16517)
2022-04-01 16:53:07 +02:00
encoder_decoder
set eos_token_id to None to generate until max length (#16989)
2022-04-28 19:47:38 +02:00
extended
Fix issue probably-meant-fstring found at https://codereview.doctor (#16913)
2022-04-25 15:15:00 -04:00
fixtures
add a warning in SpmConverter for sentencepiece's model using the byte fallback feature (#16629)
2022-04-11 11:06:10 +02:00
flaubert
Use random_attention_mask for TF tests (#16517)
2022-04-01 16:53:07 +02:00
fnet
Enable more test_torchscript (#16679)
2022-04-11 18:23:35 +02:00
fsmt
Reorganize file utils (#16264)
2022-03-23 10:26:33 -04:00
funnel
Reduce Funnel PT/TF diff (#16744)
2022-04-13 17:19:52 +02:00
generation
TF: XLA bad words logits processor and list of processors (#16974)
2022-04-29 15:54:58 +01:00
glpn
Enable more test_torchscript (#16679)
2022-04-11 18:23:35 +02:00
gpt2
TF: XLA stable softmax (#16892)
2022-04-25 20:10:51 +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
[Test refactor 1/5] Per-folder tests reorganization (#15725)
2022-02-23 15:46:28 -05:00
hubert
use scale=1.0 in floats_tensor called in speech model testers (#17007)
2022-04-29 14:41:33 +02:00
ibert
[Test refactor 1/5] Per-folder tests reorganization (#15725)
2022-02-23 15:46:28 -05:00
imagegpt
Return input_ids in ImageGPT feature extractor (#16872)
2022-04-21 09:09:00 -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
Improve PT/TF equivalence test (#16557)
2022-04-11 22:19:12 +02:00
longformer
Some tests misusing assertTrue for comparisons fix (#16771)
2022-04-19 14:44:08 +02:00
luke
[Test refactor 1/5] Per-folder tests reorganization (#15725)
2022-02-23 15:46:28 -05:00
lxmert
Improve test_pt_tf_model_equivalence on PT side (#16731)
2022-04-19 21:13:27 +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
Enable more test_torchscript (#16679)
2022-04-11 18:23:35 +02:00
mbart
Reorganize file utils (#16264)
2022-03-23 10:26:33 -04: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
Use random_attention_mask for TF tests (#16517)
2022-04-01 16:53:07 +02:00
mpnet
Enable more test_torchscript (#16679)
2022-04-11 18:23:35 +02: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 onnx config for RoFormer (#16861)
2022-04-26 16:51:15 +02:00
openai
Use random_attention_mask for TF tests (#16517)
2022-04-01 16:53:07 +02:00
optimization
[Test refactor 1/5] Per-folder tests reorganization (#15725)
2022-02-23 15:46:28 -05:00
pegasus
Reorganize file utils (#16264)
2022-03-23 10:26:33 -04:00
perceiver
use scale=1.0 in floats_tensor called in speech model testers (#17007)
2022-04-29 14:41:33 +02:00
phobert
[Test refactor 1/5] Per-folder tests reorganization (#15725)
2022-02-23 15:46:28 -05:00
pipelines
Long QuestionAnsweringPipeline fix. (#16778)
2022-04-21 09:59:25 +02: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
Enable more test_torchscript (#16679)
2022-04-11 18:23:35 +02:00
qdqbert
[Test refactor 1/5] Per-folder tests reorganization (#15725)
2022-02-23 15:46:28 -05:00
rag
Fix PyTorch RAG tests GPU OOM (#16881)
2022-04-25 17:33:56 +02:00
realm
Fix and improve REALM fine-tuning (#15297)
2022-03-03 14:10:15 +01:00
reformer
Reorganize file utils (#16264)
2022-03-23 10:26:33 -04:00
regnet
Enable more test_torchscript (#16679)
2022-04-11 18:23:35 +02:00
rembert
Use random_attention_mask for TF tests (#16517)
2022-04-01 16:53:07 +02:00
resnet
Enable more test_torchscript (#16679)
2022-04-11 18:23:35 +02: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
Clean up semantic segmentation tests (#16801)
2022-04-19 09:02:19 +02:00
sew
use scale=1.0 in floats_tensor called in speech model testers (#17007)
2022-04-29 14:41:33 +02:00
sew_d
use scale=1.0 in floats_tensor called in speech model testers (#17007)
2022-04-29 14:41:33 +02:00
speech_encoder_decoder
use scale=1.0 in floats_tensor called in speech model testers (#17007)
2022-04-29 14:41:33 +02:00
speech_to_text
Enable more test_torchscript (#16679)
2022-04-11 18:23:35 +02: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
Enable more test_torchscript (#16679)
2022-04-11 18:23:35 +02:00
swin
Enable more test_torchscript (#16679)
2022-04-11 18:23:35 +02:00
t5
TF: XLA stable softmax (#16892)
2022-04-25 20:10:51 +01:00
tapas
Enable more test_torchscript (#16679)
2022-04-11 18:23:35 +02:00
tapex
Add TAPEX (#16473)
2022-04-08 10:57:51 +02:00
tokenization
[Test refactor 1/5] Per-folder tests reorganization (#15725)
2022-02-23 15:46:28 -05:00
trainer
Add support for bitsandbytes (#15622)
2022-04-19 16:01:29 -04:00
transfo_xl
Enable more test_torchscript (#16679)
2022-04-11 18:23:35 +02:00
trocr
[Test refactor 1/5] Per-folder tests reorganization (#15725)
2022-02-23 15:46:28 -05:00
unispeech
use scale=1.0 in floats_tensor called in speech model testers (#17007)
2022-04-29 14:41:33 +02:00
unispeech_sat
use scale=1.0 in floats_tensor called in speech model testers (#17007)
2022-04-29 14:41:33 +02:00
utils
TF: Add sigmoid activation function (#16819)
2022-04-19 16:13:08 +01:00
van
Enable more test_torchscript (#16679)
2022-04-11 18:23:35 +02:00
vilt
Reorganize file utils (#16264)
2022-03-23 10:26:33 -04:00
vision_encoder_decoder
set eos_token_id to None to generate until max length (#16989)
2022-04-28 19:47:38 +02:00
vision_text_dual_encoder
Reorganize file utils (#16264)
2022-03-23 10:26:33 -04:00
visual_bert
[Test refactor 1/5] Per-folder tests reorganization (#15725)
2022-02-23 15:46:28 -05:00
vit
Clean up vision tests (#17024)
2022-05-02 16:28:58 +02:00
vit_mae
Improve test_pt_tf_model_equivalence on PT side (#16731)
2022-04-19 21:13:27 +02:00
wav2vec2
use scale=1.0 in floats_tensor called in speech model testers (#17007)
2022-04-29 14:41:33 +02:00
wav2vec2_phoneme
Some tests misusing assertTrue for comparisons fix (#16771)
2022-04-19 14:44:08 +02:00
wav2vec2_with_lm
Some tests misusing assertTrue for comparisons fix (#16771)
2022-04-19 14:44:08 +02:00
wavlm
use scale=1.0 in floats_tensor called in speech model testers (#17007)
2022-04-29 14:41:33 +02:00
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
[Test refactor 1/5] Per-folder tests reorganization (#15725)
2022-02-23 15:46:28 -05:00
xlnet
Use random_attention_mask for TF tests (#16517)
2022-04-01 16:53:07 +02:00
yolos
Add YOLOS (#16848)
2022-05-02 18:30:55 +02: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
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
Improve test_pt_tf_model_equivalence on PT side (#16731)
2022-04-19 21:13:27 +02:00
test_modeling_flax_common.py
[Flax] improve large model init and loading (#16148)
2022-04-19 14:19:55 +02:00
test_modeling_tf_common.py
TF: XLA stable softmax (#16892)
2022-04-25 20:10:51 +01:00
test_sequence_feature_extraction_common.py
Some tests misusing assertTrue for comparisons fix (#16771)
2022-04-19 14:44:08 +02:00
test_tokenization_common.py
Fix #16660 (tokenizers setters of ids of special tokens) (#16661)
2022-04-13 07:49:06 -04:00