SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity
Files
5b72b3412b8efb3177d2daf90798e2ee20cb21e9
HuggingFace_transformer/tests/models
History
Quentin Meeus 5b72b3412b Remove CLI spams with Whisper FeatureExtractor (#21267) 
* Remove CLI spams with Whisper FeatureExtractor

Whisper feature extractor representation includes the MEL filters, a list of list that is represented as ~16,000 lines. This needlessly spams the command line. I added a `__repr__` method that replaces this list with a string "<array of shape (80, 201)>"

* Remove mel_filters from to_dict output  

Credits to @ArthurZucker

* remove unused import

* update feature extraction tests for the changes in to_dict
2023-02-10 09:15:16 -05:00
..
albert
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
altclip
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
audio_spectrogram_transformer
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
auto
[tokenizer] sanitize saved config (#21483)
2023-02-07 10:51:45 -08:00
bart
🚨🚨🚨 Enforce single model initialization (#21431)
2023-02-09 15:46:26 -05: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
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
bert
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
bert_generation
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05: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
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
bigbird_pegasus
Generate: move generation_*.py src files into generation/*.py (#20096)
2022-11-09 15:34:08 +00:00
biogpt
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
bit
[Backbones] Improve out features (#20675)
2022-12-09 09:14:52 +01:00
blenderbot
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
blenderbot_small
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
blip
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
blip_2
Add BLIP-2 (#21441)
2023-02-09 16:52:11 +01:00
bloom
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
bort
Move test model folders (#17034)
2022-05-03 14:42:02 +02:00
bridgetower
Tag tests as slow ⌛ (#21537)
2023-02-09 14:46:15 +00:00
byt5
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
camembert
Added with torch.no_grad() to Camembert integration test (#21544)
2023-02-10 10:58:29 +01:00
canine
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
chinese_clip
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
clip
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
clipseg
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
codegen
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
conditional_detr
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
convbert
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
convnext
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
cpm
Move test model folders (#17034)
2022-05-03 14:42:02 +02:00
ctrl
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
cvt
[CvT] Tensorflow implementation (#18597)
2022-10-11 18:16:52 +01:00
data2vec
update relative positional embedding (#20203)
2022-11-15 10:46:34 +01:00
deberta
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
deberta_v2
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
decision_transformer
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
deformable_detr
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
deit
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
deta
🚨🚨🚨 Enforce single model initialization (#21431)
2023-02-09 15:46:26 -05:00
detr
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
dinat
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
distilbert
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
dit
Move test model folders (#17034)
2022-05-03 14:42:02 +02:00
donut
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
dpr
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
dpt
🚨🚨🚨 Enforce single model initialization (#21431)
2023-02-09 15:46:26 -05:00
efficientformer
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
electra
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
encoder_decoder
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
ernie
update relative positional embedding (#20203)
2022-11-15 10:46:34 +01:00
esm
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
flaubert
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
flava
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
fnet
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
fsmt
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
funnel
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
git
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
glpn
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
gpt2
[OPT] Adds GPT2TokenizerFast to the list of tokenizer to use for OPT. (#20823)
2023-02-07 17:35:28 +01:00
gpt_neo
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
gpt_neox
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
gpt_neox_japanese
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
gpt_sw3
Add gpt-sw3 model to transformers (#20209)
2022-12-12 13:12:13 -05:00
gptj
[CI ] Remove past in favor of pat_key_values (#21443)
2023-02-07 09:51:35 +01:00
graphormer
Fix Graphormer test suite (#21419)
2023-02-02 16:29:13 +01:00
groupvit
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
herbert
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
hubert
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
ibert
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
imagegpt
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
jukebox
Ci-jukebox (#20613)
2022-12-06 16:14:03 +01:00
layoutlm
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
layoutlmv2
🚨🚨🚨 Enforce single model initialization (#21431)
2023-02-09 15:46:26 -05:00
layoutlmv3
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
layoutxlm
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
led
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
levit
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
lilt
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
longformer
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
longt5
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
luke
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
lxmert
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
m2m_100
Exclude the madeup words from M2M100Tokenizer.vocab_size (#20976)
2023-02-08 09:19:06 -05:00
marian
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
markuplm
[processor] Add 'model input names' property (#20117)
2022-11-10 19:29:20 +00:00
mask2former
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
maskformer
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
mbart
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
mbart50
Replace as_target context managers by direct calls (#18325)
2022-07-29 08:09:09 -04:00
mctct
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
megatron_bert
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
megatron_gpt2
Move test model folders (#17034)
2022-05-03 14:42:02 +02:00
mluke
Black preview (#17217)
2022-05-12 16:25:55 -04:00
mobilebert
Tag tests as slow ⌛ (#21537)
2023-02-09 14:46:15 +00:00
mobilenet_v1
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
mobilenet_v2
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
mobilevit
Tag tests as slow ⌛ (#21537)
2023-02-09 14:46:15 +00:00
mpnet
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
mt5
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
mvp
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
nat
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
nezha
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
nllb
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
nystromformer
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
oneformer
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
openai
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
opt
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
owlvit
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
pegasus
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
pegasus_x
Generate: move generation_*.py src files into generation/*.py (#20096)
2022-11-09 15:34:08 +00:00
perceiver
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
phobert
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
plbart
Generate: fix plbart generation tests (#20391)
2022-11-22 17:56:04 +00:00
poolformer
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
prophetnet
🚨🚨🚨 Enforce single model initialization (#21431)
2023-02-09 15:46:26 -05:00
qdqbert
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
rag
Tag tests as slow ⌛ (#21537)
2023-02-09 14:46:15 +00:00
realm
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
reformer
🚨🚨🚨 Enforce single model initialization (#21431)
2023-02-09 15:46:26 -05:00
regnet
[Tests] Improve test_attention_outputs (#20701)
2022-12-14 14:41:40 +01:00
rembert
Wrap RemBert integration test forward passes with torch.no_grad() (#21503)
2023-02-08 14:00:52 +01:00
resnet
[Tests] Improve test_attention_outputs (#20701)
2022-12-14 14:41:40 +01:00
retribert
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
roberta
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
roberta_prelayernorm
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
roc_bert
fix both failing RoCBert tests (#20469)
2022-11-28 17:08:57 +01:00
roformer
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
segformer
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
sew
Move test model folders (#17034)
2022-05-03 14:42:02 +02:00
sew_d
Move test model folders (#17034)
2022-05-03 14:42:02 +02:00
speech_encoder_decoder
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
speech_to_text
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
speech_to_text_2
Generate: move generation_*.py src files into generation/*.py (#20096)
2022-11-09 15:34:08 +00:00
speecht5
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
splinter
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
squeezebert
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
swin
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
swin2sr
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
swinv2
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
switch_transformers
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
t5
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
table_transformer
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
tapas
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
tapex
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
time_series_transformer
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
timesformer
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
trajectory_transformer
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
transfo_xl
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
trocr
Generate: move generation_*.py src files into generation/*.py (#20096)
2022-11-09 15:34:08 +00:00
unispeech
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
unispeech_sat
Move test model folders (#17034)
2022-05-03 14:42:02 +02:00
upernet
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
van
[Tests] Improve test_attention_outputs (#20701)
2022-12-14 14:41:40 +01:00
videomae
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
vilt
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
vision_encoder_decoder
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
vision_text_dual_encoder
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
visual_bert
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
vit
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
vit_hybrid
🚨🚨🚨 Enforce single model initialization (#21431)
2023-02-09 15:46:26 -05:00
vit_mae
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
vit_msn
Some fixes regarding auto mappings and test class names (#19923)
2022-10-27 14:38:59 +02:00
wav2vec2
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
wav2vec2_conformer
[Test] Fix W2V-Conformer integration test (#17303)
2022-05-17 18:20:36 +02:00
wav2vec2_phoneme
Move test model folders (#17034)
2022-05-03 14:42:02 +02:00
wav2vec2_with_lm
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
wavlm
Move test model folders (#17034)
2022-05-03 14:42:02 +02:00
whisper
Remove CLI spams with Whisper FeatureExtractor (#21267)
2023-02-10 09:15:16 -05:00
x_clip
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
xglm
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
xlm
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
xlm_prophetnet
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
xlm_roberta
Added with torch.no_grad() to XLM-Roberta integration test (#21547)
2023-02-09 21:49:54 +01:00
xlm_roberta_xl
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
xlnet
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
yolos
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
yoso
Update quality tooling for formatting (#21480)
2023-02-06 18:10:56 -05:00
__init__.py
Move test model folders (#17034)
2022-05-03 14:42:02 +02:00