SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity
Files
a717e0318ce4b4973cfebee44d1747f5ca828ac2
HuggingFace_transformer/tests/models
History
amyeroberts a717e0318c Add TimmBackbone model (#22619) 
* Add test_backbone for convnext

* Add TimmBackbone model

* Add check for backbone type

* Tidying up - config checks

* Update convnextv2

* Tidy up

* Fix indices & clearer comment

* Exceptions for config checks

* Correclty update config for tests

* Safer imports

* Safer safer imports

* Fix where decorators go

* Update import logic and backbone tests

* More import fixes

* Fixup

* Only import all_models if torch available

* Fix kwarg updates in from_pretrained & main rebase

* Tidy up

* Add tests for AutoBackbone

* Tidy up

* Fix import error

* Fix up

* Install nattan in doc_test_job

* Revert back to setting self._out_xxx directly

* Bug fix - out_indices mapping from out_features

* Fix tests

* Dont accept output_loading_info for Timm models

* Set out_xxx and don't remap

* Use smaller checkpoint for test

* Don't remap timm indices - check out_indices based on stage names

* Skip test as it's n/a

* Apply suggestions from code review

Co-authored-by: Sylvain Gugger <35901082+sgugger@users.noreply.github.com>

* Cleaner imports / spelling is hard

---------

Co-authored-by: Sylvain Gugger <35901082+sgugger@users.noreply.github.com>
2023-06-06 17:11:30 +01:00
..
albert
Better TF docstring types (#23477)
2023-05-24 13:52:52 +01:00
align
Skip failing AlignModelTest::test_multi_gpu_data_parallel_forward (#23374)
2023-05-15 16:46:58 +02:00
altclip
Automatically create/update tiny models (#22275)
2023-03-23 19:14:17 +01:00
audio_spectrogram_transformer
is_batched fix for remaining 2-D numpy arrays (#23309)
2023-05-23 14:37:35 -04:00
auto
Add TimmBackbone model (#22619)
2023-06-06 17:11:30 +01:00
autoformer
[Time-Series] Autoformer model (#21891)
2023-05-30 10:23:32 +02:00
bart
Better TF docstring types (#23477)
2023-05-24 13:52:52 +01:00
barthez
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
bartpho
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
beit
🔥Rework pipeline testing by removing PipelineTestCaseMeta 🚀 (#21516)
2023-02-28 19:40:57 +01:00
bert
Better TF docstring types (#23477)
2023-05-24 13:52:52 +01:00
bert_generation
🔥Rework pipeline testing by removing PipelineTestCaseMeta 🚀 (#21516)
2023-02-28 19:40:57 +01:00
bert_japanese
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
bertweet
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
big_bird
fix random attention for pytorch's bigbird/pegasus_bigbird (#23056)
2023-05-07 18:55:04 -04:00
bigbird_pegasus
Generate: skip left-padding tests on old models (#23437)
2023-05-18 11:04:51 +01:00
biogpt
Update tiny models and pipeline tests (#23446)
2023-05-18 17:29:04 +02:00
bit
Backbone add mixin tests (#22542)
2023-04-06 13:50:15 +01:00
blenderbot
Better TF docstring types (#23477)
2023-05-24 13:52:52 +01:00
blenderbot_small
Better TF docstring types (#23477)
2023-05-24 13:52:52 +01:00
blip
Better TF docstring types (#23477)
2023-05-24 13:52:52 +01:00
blip_2
[Blip2] skip accelerate test (#22124)
2023-03-13 15:03:21 +01:00
bloom
Generate: add test to check KV format (#23403)
2023-05-16 19:28:19 +01:00
bort
Better TF docstring types (#23477)
2023-05-24 13:52:52 +01:00
bridgetower
Fix a BridgeTower test (#23694)
2023-05-23 17:32:57 +02:00
byt5
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
camembert
Better TF docstring types (#23477)
2023-05-24 13:52:52 +01:00
canine
🔥Rework pipeline testing by removing PipelineTestCaseMeta 🚀 (#21516)
2023-02-28 19:40:57 +01:00
chinese_clip
🔥Rework pipeline testing by removing PipelineTestCaseMeta 🚀 (#21516)
2023-02-28 19:40:57 +01:00
clap
is_batched fix for remaining 2-D numpy arrays (#23309)
2023-05-23 14:37:35 -04:00
clip
Better TF docstring types (#23477)
2023-05-24 13:52:52 +01:00
clipseg
🔥Rework pipeline testing by removing PipelineTestCaseMeta 🚀 (#21516)
2023-02-28 19:40:57 +01:00
codegen
Decorate test_codegen_sample_max_time as flaky (#22953)
2023-04-24 15:27:31 +02:00
conditional_detr
Skip test_multi_gpu_data_parallel_forward for some model tests (#21991)
2023-03-07 14:23:36 +01:00
convbert
Better TF docstring types (#23477)
2023-05-24 13:52:52 +01:00
convnext
Better TF docstring types (#23477)
2023-05-24 13:52:52 +01:00
convnextv2
Update ConvNextV2ModelIntegrationTest::test_inference_image_classification_head (#23402)
2023-05-16 23:35:11 +02:00
cpm
Fix PipelineTests skip conditions (#22320)
2023-03-22 20:02:24 +01:00
cpmant
Update tiny models and pipeline tests (#23446)
2023-05-18 17:29:04 +02:00
ctrl
Better TF docstring types (#23477)
2023-05-24 13:52:52 +01:00
cvt
Better TF docstring types (#23477)
2023-05-24 13:52:52 +01:00
data2vec
Better TF docstring types (#23477)
2023-05-24 13:52:52 +01:00
deberta
Better TF docstring types (#23477)
2023-05-24 13:52:52 +01:00
deberta_v2
Better TF docstring types (#23477)
2023-05-24 13:52:52 +01:00
decision_transformer
🔥Rework pipeline testing by removing PipelineTestCaseMeta 🚀 (#21516)
2023-02-28 19:40:57 +01:00
deformable_detr
🔥Rework pipeline testing by removing PipelineTestCaseMeta 🚀 (#21516)
2023-02-28 19:40:57 +01:00
deit
Better TF docstring types (#23477)
2023-05-24 13:52:52 +01:00
deta
Revert (for now) the change on Deta in #22437 (#22750)
2023-04-13 21:32:29 +02:00
detr
[DETR and friends] Remove is_timm_available (#21814)
2023-03-07 15:19:39 -05:00
dinat
Backbone add mixin tests (#22542)
2023-04-06 13:50:15 +01:00
distilbert
Better TF docstring types (#23477)
2023-05-24 13:52:52 +01:00
dit
Move test model folders (#17034)
2022-05-03 14:42:02 +02:00
donut
🔥Rework pipeline testing by removing PipelineTestCaseMeta 🚀 (#21516)
2023-02-28 19:40:57 +01:00
dpr
Better TF docstring types (#23477)
2023-05-24 13:52:52 +01:00
dpt
🔥Rework pipeline testing by removing PipelineTestCaseMeta 🚀 (#21516)
2023-02-28 19:40:57 +01:00
efficientformer
Add TensorFlow implementation of EfficientFormer (#22620)
2023-05-31 10:43:12 +01:00
efficientnet
Update tiny model summary file for recent models (#22637)
2023-04-06 22:52:59 +02:00
electra
Better TF docstring types (#23477)
2023-05-24 13:52:52 +01:00
encoder_decoder
Better TF docstring types (#23477)
2023-05-24 13:52:52 +01:00
ernie
🔥Rework pipeline testing by removing PipelineTestCaseMeta 🚀 (#21516)
2023-02-28 19:40:57 +01:00
ernie_m
Automatically create/update tiny models (#22275)
2023-03-23 19:14:17 +01:00
esm
Better TF docstring types (#23477)
2023-05-24 13:52:52 +01:00
flaubert
Better TF docstring types (#23477)
2023-05-24 13:52:52 +01:00
flava
🔥Rework pipeline testing by removing PipelineTestCaseMeta 🚀 (#21516)
2023-02-28 19:40:57 +01:00
fnet
Remove some pipeline skip cases (#22865)
2023-04-19 20:27:19 +02:00
focalnet
Update tiny models and pipeline tests (#23446)
2023-05-18 17:29:04 +02:00
fsmt
update_pip_test_mapping (#22606)
2023-04-06 17:56:06 +02:00
funnel
Better TF docstring types (#23477)
2023-05-24 13:52:52 +01:00
git
🔥Rework pipeline testing by removing PipelineTestCaseMeta 🚀 (#21516)
2023-02-28 19:40:57 +01:00
glpn
🔥Rework pipeline testing by removing PipelineTestCaseMeta 🚀 (#21516)
2023-02-28 19:40:57 +01:00
gpt2
Better TF docstring types (#23477)
2023-05-24 13:52:52 +01:00
gpt_bigcode
Generate: add test to check KV format (#23403)
2023-05-16 19:28:19 +01:00
gpt_neo
Update tiny models and pipeline tests (#23446)
2023-05-18 17:29:04 +02:00
gpt_neox
Update tiny models and pipeline tests (#23446)
2023-05-18 17:29:04 +02:00
gpt_neox_japanese
🔥Rework pipeline testing by removing PipelineTestCaseMeta 🚀 (#21516)
2023-02-28 19:40:57 +01:00
gpt_sw3
Add gpt-sw3 model to transformers (#20209)
2022-12-12 13:12:13 -05:00
gptj
Better TF docstring types (#23477)
2023-05-24 13:52:52 +01:00
gptsan_japanese
Update tiny model summary file for recent models (#22637)
2023-04-06 22:52:59 +02:00
graphormer
🔥Rework pipeline testing by removing PipelineTestCaseMeta 🚀 (#21516)
2023-02-28 19:40:57 +01:00
groupvit
Better TF docstring types (#23477)
2023-05-24 13:52:52 +01:00
herbert
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
hubert
Better TF docstring types (#23477)
2023-05-24 13:52:52 +01:00
ibert
🔥Rework pipeline testing by removing PipelineTestCaseMeta 🚀 (#21516)
2023-02-28 19:40:57 +01:00
imagegpt
Generate: skip left-padding tests on old models (#23437)
2023-05-18 11:04:51 +01:00
informer
[Time-Series] Autoformer model (#21891)
2023-05-30 10:23:32 +02:00
jukebox
Update Jukebox tests (#21984)
2023-03-07 04:20:14 +01:00
layoutlm
Better TF docstring types (#23477)
2023-05-24 13:52:52 +01:00
layoutlmv2
Don't use LayoutLMv2 and LayoutLMv3 in some pipeline tests (#22774)
2023-04-17 17:45:20 +02:00
layoutlmv3
Better TF docstring types (#23477)
2023-05-24 13:52:52 +01:00
layoutxlm
Apply ruff flake8-comprehensions (#21694)
2023-02-22 09:14:54 +01:00
led
Better TF docstring types (#23477)
2023-05-24 13:52:52 +01:00
levit
🔥Rework pipeline testing by removing PipelineTestCaseMeta 🚀 (#21516)
2023-02-28 19:40:57 +01:00
lilt
Move is_pipeline_test_to_skip to specific model test classes (#21999)
2023-03-14 10:03:02 +01:00
llama
[LlamaTokenizerFast] nit update post_processor on the fly (#23855)
2023-05-30 16:50:41 +02:00
longformer
Better TF docstring types (#23477)
2023-05-24 13:52:52 +01:00
longt5
update_pip_test_mapping (#22606)
2023-04-06 17:56:06 +02:00
luke
Fix PipelineTests skip conditions (#22320)
2023-03-22 20:02:24 +01:00
lxmert
Better TF docstring types (#23477)
2023-05-24 13:52:52 +01:00
m2m_100
update_pip_test_mapping (#22606)
2023-04-06 17:56:06 +02:00
marian
Better TF docstring types (#23477)
2023-05-24 13:52:52 +01:00
markuplm
Update some MarkupLM tests' expected values (#22667)
2023-04-11 10:00:34 +02:00
mask2former
Fix quality due to ruff release
2023-03-22 20:45:08 -04:00
maskformer
Backbone add mixin tests (#22542)
2023-04-06 13:50:15 +01:00
mbart
Better TF docstring types (#23477)
2023-05-24 13:52:52 +01:00
mbart50
Replace as_target context managers by direct calls (#18325)
2022-07-29 08:09:09 -04:00
mctct
is_batched fix for remaining 2-D numpy arrays (#23309)
2023-05-23 14:37:35 -04:00
mega
Fix MegaModel CI (#22652)
2023-04-07 17:13:04 +02:00
megatron_bert
🔥Rework pipeline testing by removing PipelineTestCaseMeta 🚀 (#21516)
2023-02-28 19:40:57 +01:00
megatron_gpt2
Move test model folders (#17034)
2022-05-03 14:42:02 +02:00
mgp_str
update_pip_test_mapping (#22606)
2023-04-06 17:56:06 +02:00
mluke
Black preview (#17217)
2022-05-12 16:25:55 -04:00
mobilebert
Better TF docstring types (#23477)
2023-05-24 13:52:52 +01:00
mobilenet_v1
🔥Rework pipeline testing by removing PipelineTestCaseMeta 🚀 (#21516)
2023-02-28 19:40:57 +01:00
mobilenet_v2
🔥Rework pipeline testing by removing PipelineTestCaseMeta 🚀 (#21516)
2023-02-28 19:40:57 +01:00
mobilevit
Better TF docstring types (#23477)
2023-05-24 13:52:52 +01:00
mobilevitv2
Fix MobileViTV2 checkpoint name (#24018)
2023-06-05 18:12:45 +02:00
mpnet
Better TF docstring types (#23477)
2023-05-24 13:52:52 +01:00
mt5
Better TF docstring types (#23477)
2023-05-24 13:52:52 +01:00
mvp
Generate: skip left-padding tests on old models (#23437)
2023-05-18 11:04:51 +01:00
nat
Backbone add mixin tests (#22542)
2023-04-06 13:50:15 +01:00
nezha
🔥Rework pipeline testing by removing PipelineTestCaseMeta 🚀 (#21516)
2023-02-28 19:40:57 +01:00
nllb
🚨🚨🚨 [NLLB Tokenizer] Fix the prefix tokens 🚨🚨🚨 (#22313)
2023-04-04 14:53:06 +02:00
nllb_moe
tests: Fix flaky test for NLLB-MoE (#22880)
2023-04-21 17:09:40 +01:00
nystromformer
🔥Rework pipeline testing by removing PipelineTestCaseMeta 🚀 (#21516)
2023-02-28 19:40:57 +01:00
oneformer
Automatically create/update tiny models (#22275)
2023-03-23 19:14:17 +01:00
open_llama
add open-llama model with ckpt (#22795)
2023-04-28 11:01:32 -04:00
openai
Better TF docstring types (#23477)
2023-05-24 13:52:52 +01:00
opt
Better TF docstring types (#23477)
2023-05-24 13:52:52 +01:00
owlvit
🔥Rework pipeline testing by removing PipelineTestCaseMeta 🚀 (#21516)
2023-02-28 19:40:57 +01:00
pegasus
Better TF docstring types (#23477)
2023-05-24 13:52:52 +01:00
pegasus_x
update_pip_test_mapping (#22606)
2023-04-06 17:56:06 +02:00
perceiver
🔥Rework pipeline testing by removing PipelineTestCaseMeta 🚀 (#21516)
2023-02-28 19:40:57 +01:00
phobert
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
pix2struct
Pix2Struct: fix wrong broadcast axis of attention mask in visual encoder (#23976)
2023-06-05 11:47:29 -04:00
plbart
Generate: skip left-padding tests on old models (#23437)
2023-05-18 11:04:51 +01:00
poolformer
🔥Rework pipeline testing by removing PipelineTestCaseMeta 🚀 (#21516)
2023-02-28 19:40:57 +01:00
prophetnet
Generate: skip left-padding tests on old models (#23437)
2023-05-18 11:04:51 +01:00
qdqbert
🔥Rework pipeline testing by removing PipelineTestCaseMeta 🚀 (#21516)
2023-02-28 19:40:57 +01:00
rag
Better TF docstring types (#23477)
2023-05-24 13:52:52 +01:00
realm
Unpin numba (#23162)
2023-05-31 14:59:30 +01:00
reformer
Generate: skip left-padding tests on old models (#23437)
2023-05-18 11:04:51 +01:00
regnet
Better TF docstring types (#23477)
2023-05-24 13:52:52 +01:00
rembert
Better TF docstring types (#23477)
2023-05-24 13:52:52 +01:00
resnet
Better TF docstring types (#23477)
2023-05-24 13:52:52 +01:00
retribert
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
roberta
Better TF docstring types (#23477)
2023-05-24 13:52:52 +01:00
roberta_prelayernorm
Better TF docstring types (#23477)
2023-05-24 13:52:52 +01:00
roc_bert
Move is_pipeline_test_to_skip to specific model test classes (#21999)
2023-03-14 10:03:02 +01:00
roformer
Better TF docstring types (#23477)
2023-05-24 13:52:52 +01:00
rwkv
Update tiny models and pipeline tests (#23446)
2023-05-18 17:29:04 +02:00
sam
Better TF docstring types (#23477)
2023-05-24 13:52:52 +01:00
segformer
Better TF docstring types (#23477)
2023-05-24 13:52:52 +01:00
sew
🔥Rework pipeline testing by removing PipelineTestCaseMeta 🚀 (#21516)
2023-02-28 19:40:57 +01:00
sew_d
🔥Rework pipeline testing by removing PipelineTestCaseMeta 🚀 (#21516)
2023-02-28 19:40:57 +01:00
speech_encoder_decoder
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
speech_to_text
Better TF docstring types (#23477)
2023-05-24 13:52:52 +01:00
speech_to_text_2
🔥Rework pipeline testing by removing PipelineTestCaseMeta 🚀 (#21516)
2023-02-28 19:40:57 +01:00
speecht5
is_batched fix for remaining 2-D numpy arrays (#23309)
2023-05-23 14:37:35 -04:00
splinter
Make tiny model creation + pipeline testing more robust (#22500)
2023-04-06 17:45:55 +02:00
squeezebert
🔥Rework pipeline testing by removing PipelineTestCaseMeta 🚀 (#21516)
2023-02-28 19:40:57 +01:00
swiftformer
Update tiny models and pipeline tests (#23446)
2023-05-18 17:29:04 +02:00
swin
Better TF docstring types (#23477)
2023-05-24 13:52:52 +01:00
swin2sr
Skip test_multi_gpu_data_parallel_forward for some model tests (#21991)
2023-03-07 14:23:36 +01:00
swinv2
Skip test_multi_gpu_data_parallel_forward for some model tests (#21991)
2023-03-07 14:23:36 +01:00
switch_transformers
update_pip_test_mapping (#22606)
2023-04-06 17:56:06 +02:00
t5
Better TF docstring types (#23477)
2023-05-24 13:52:52 +01:00
table_transformer
🔥Rework pipeline testing by removing PipelineTestCaseMeta 🚀 (#21516)
2023-02-28 19:40:57 +01:00
tapas
Better TF docstring types (#23477)
2023-05-24 13:52:52 +01:00
tapex
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
time_series_transformer
[Time-Series] Autoformer model (#21891)
2023-05-30 10:23:32 +02:00
timesformer
🔥Rework pipeline testing by removing PipelineTestCaseMeta 🚀 (#21516)
2023-02-28 19:40:57 +01:00
timm_backbone
Add TimmBackbone model (#22619)
2023-06-06 17:11:30 +01:00
trajectory_transformer
Time to Say Goodbye, torch 1.7 and 1.8 (#22291)
2023-03-21 19:22:01 +01:00
transfo_xl
Better TF docstring types (#23477)
2023-05-24 13:52:52 +01:00
trocr
Generate: skip left-padding tests on old models (#23437)
2023-05-18 11:04:51 +01:00
tvlt
is_batched fix for remaining 2-D numpy arrays (#23309)
2023-05-23 14:37:35 -04:00
unispeech
🔥Rework pipeline testing by removing PipelineTestCaseMeta 🚀 (#21516)
2023-02-28 19:40:57 +01:00
unispeech_sat
🔥Rework pipeline testing by removing PipelineTestCaseMeta 🚀 (#21516)
2023-02-28 19:40:57 +01:00
upernet
🔥Rework pipeline testing by removing PipelineTestCaseMeta 🚀 (#21516)
2023-02-28 19:40:57 +01:00
van
🔥Rework pipeline testing by removing PipelineTestCaseMeta 🚀 (#21516)
2023-02-28 19:40:57 +01:00
videomae
🔥Rework pipeline testing by removing PipelineTestCaseMeta 🚀 (#21516)
2023-02-28 19:40:57 +01:00
vilt
🔥Rework pipeline testing by removing PipelineTestCaseMeta 🚀 (#21516)
2023-02-28 19:40:57 +01:00
vision_encoder_decoder
Better TF docstring types (#23477)
2023-05-24 13:52:52 +01:00
vision_text_dual_encoder
Better TF docstring types (#23477)
2023-05-24 13:52:52 +01:00
visual_bert
🔥Rework pipeline testing by removing PipelineTestCaseMeta 🚀 (#21516)
2023-02-28 19:40:57 +01:00
vit
Better TF docstring types (#23477)
2023-05-24 13:52:52 +01:00
vit_hybrid
🔥Rework pipeline testing by removing PipelineTestCaseMeta 🚀 (#21516)
2023-02-28 19:40:57 +01:00
vit_mae
Better TF docstring types (#23477)
2023-05-24 13:52:52 +01:00
vit_msn
🔥Rework pipeline testing by removing PipelineTestCaseMeta 🚀 (#21516)
2023-02-28 19:40:57 +01:00
wav2vec2
[MMS] Scaling Speech Technology to 1,000+ Languages | Add attention adapter to Wav2Vec2 (#23813)
2023-06-02 10:30:24 +01:00
wav2vec2_conformer
🔥Rework pipeline testing by removing PipelineTestCaseMeta 🚀 (#21516)
2023-02-28 19:40:57 +01:00
wav2vec2_phoneme
Move test model folders (#17034)
2022-05-03 14:42:02 +02:00
wav2vec2_with_lm
Fix test_word_time_stamp_integration for Wav2Vec2ProcessorWithLMTest (#22800)
2023-04-17 12:41:55 +02:00
wavlm
🔥Rework pipeline testing by removing PipelineTestCaseMeta 🚀 (#21516)
2023-02-28 19:40:57 +01:00
whisper
[Whisper Tokenizer] Skip special tokens when decoding with timestamps (#23945)
2023-06-02 16:26:59 +02:00
x_clip
🔥Rework pipeline testing by removing PipelineTestCaseMeta 🚀 (#21516)
2023-02-28 19:40:57 +01:00
xglm
Better TF docstring types (#23477)
2023-05-24 13:52:52 +01:00
xlm
Better TF docstring types (#23477)
2023-05-24 13:52:52 +01:00
xlm_prophetnet
Update expected values in XLMProphetNetModelIntegrationTest (#21957)
2023-03-06 09:15:44 +01:00
xlm_roberta
Better TF docstring types (#23477)
2023-05-24 13:52:52 +01:00
xlm_roberta_xl
Use real tokenizers if tiny version(s) creation has issue(s) (#22428)
2023-03-29 16:16:23 +02:00
xlnet
Better TF docstring types (#23477)
2023-05-24 13:52:52 +01:00
xmod
Use real tokenizers if tiny version(s) creation has issue(s) (#22428)
2023-03-29 16:16:23 +02:00
yolos
[DETR, YOLOS] Fix device bug (#21974)
2023-03-07 07:34:04 -05:00
yoso
🔥Rework pipeline testing by removing PipelineTestCaseMeta 🚀 (#21516)
2023-02-28 19:40:57 +01:00
__init__.py
Move test model folders (#17034)
2022-05-03 14:42:02 +02:00