SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity
Files
f6261d7d81edd036fc53bfede65fe91f01a661aa
HuggingFace_transformer/tests/models
History
Yoach Lacombe c43b380e70 Add MusicGen Melody (#28819) 
* first modeling code

* make repository

* still WIP

* update model

* add tests

* add latest change

* clean docstrings and copied from

* update docstrings md and readme

* correct chroma function

* correct copied from and remove unreleated test

* add doc to toctree

* correct imports

* add convert script to notdoctested

* Add suggestion from Sanchit

Co-authored-by: Sanchit Gandhi <93869735+sanchit-gandhi@users.noreply.github.com>

* correct get_uncoditional_inputs docstrings

* modify README according to SANCHIT feedback

* add chroma to audio utils

* clean librosa and torchaudio hard dependencies

* fix FE

* refactor audio decoder -> audio encoder for consistency with previous musicgen

* refactor conditional -> encoder

* modify sampling rate logics

* modify license at the beginning

* refactor all_self_attns->all_attentions

* remove ignore copy from causallm generate

* add copied from for from_sub_models

* fix make copies

* add warning if audio is truncated

* add copied from where relevant

* remove artefact

* fix convert script

* fix torchaudio and FE

* modify chroma method according to feedback-> better naming

* refactor input_values->input_features

* refactor input_values->input_features and fix import fe

* add input_features to docstrigs

* correct inputs_embeds logics

* remove dtype conversion

* refactor _prepare_conditional_hidden_states_kwargs_for_generation ->_prepare_encoder_hidden_states_kwargs_for_generation

* change warning for chroma length

* Update src/transformers/models/musicgen_melody/convert_musicgen_melody_transformers.py

Co-authored-by: Sanchit Gandhi <93869735+sanchit-gandhi@users.noreply.github.com>

* change way to save wav, using soundfile

* correct docs and change to soundfile

* fix import

* fix init proj layers

* remove line breaks from md

* fix issue with docstrings

* add FE suggestions

* improve is in logics and remove useless imports

* remove custom from_pretrained

* simplify docstring code

* add suggestions for modeling tests

* make style

* update converting script with sanity check

* remove encoder attention mask from conditional generation

* replace musicgen melody checkpoints with official orga

* rename ylacombe->facebook in checkpoints

* fix copies

* remove unecessary warning

* add shape in code docstrings

* add files to slow doc tests

* fix md bug and add md to not_tested

* make fix-copies

* fix hidden states test and batching

---------

Co-authored-by: Sanchit Gandhi <93869735+sanchit-gandhi@users.noreply.github.com>
2024-03-18 13:06:12 +00:00
..
albert
Adds pretrained IDs directly in the tests (#29534)
2024-03-13 14:53:27 +01:00
align
Add tests for batching support (#29297)
2024-03-12 17:46:19 +00:00
altclip
Add tests for batching support (#29297)
2024-03-12 17:46:19 +00:00
audio_spectrogram_transformer
Add numpy alternative to FE using torchaudio (#26339)
2023-11-08 07:39:37 +00:00
auto
Allow remote code repo names to contain "." (#29175)
2024-02-23 12:46:31 +00:00
autoformer
Add tests for batching support (#29297)
2024-03-12 17:46:19 +00:00
bark
Add tests for batching support (#29297)
2024-03-12 17:46:19 +00:00
bart
Adds pretrained IDs directly in the tests (#29534)
2024-03-13 14:53:27 +01:00
barthez
Adds pretrained IDs directly in the tests (#29534)
2024-03-13 14:53:27 +01:00
bartpho
Adds pretrained IDs directly in the tests (#29534)
2024-03-13 14:53:27 +01:00
beit
Avoid using uncessary get_values(MODEL_MAPPING) (#29362)
2024-02-29 17:19:17 +08:00
bert
Adds pretrained IDs directly in the tests (#29534)
2024-03-13 14:53:27 +01:00
bert_generation
Adds pretrained IDs directly in the tests (#29534)
2024-03-13 14:53:27 +01:00
bert_japanese
Adds pretrained IDs directly in the tests (#29534)
2024-03-13 14:53:27 +01:00
bertweet
Adds pretrained IDs directly in the tests (#29534)
2024-03-13 14:53:27 +01:00
big_bird
Adds pretrained IDs directly in the tests (#29534)
2024-03-13 14:53:27 +01:00
bigbird_pegasus
Generate: left-padding test, revisited (#29515)
2024-03-08 10:06:46 +00:00
biogpt
Adds pretrained IDs directly in the tests (#29534)
2024-03-13 14:53:27 +01:00
bit
Image Feature Extraction pipeline (#28216)
2024-02-05 14:50:07 +00:00
blenderbot
Generate: left-padding test, revisited (#29515)
2024-03-08 10:06:46 +00:00
blenderbot_small
Adds pretrained IDs directly in the tests (#29534)
2024-03-13 14:53:27 +01:00
blip
Add tests for batching support (#29297)
2024-03-12 17:46:19 +00:00
blip_2
Add tests for batching support (#29297)
2024-03-12 17:46:19 +00:00
bloom
Adds pretrained IDs directly in the tests (#29534)
2024-03-13 14:53:27 +01:00
bridgetower
Refactor image processor testers (#25450)
2023-08-11 11:30:18 +01:00
bros
Update tiny model information and pipeline tests (#26285)
2023-09-25 18:08:12 +02:00
byt5
[Docs] Fix spelling and grammar mistakes (#28825)
2024-02-02 08:45:00 +01:00
camembert
Adds pretrained IDs directly in the tests (#29534)
2024-03-13 14:53:27 +01:00
canine
Adds pretrained IDs directly in the tests (#29534)
2024-03-13 14:53:27 +01:00
chinese_clip
Add tests for batching support (#29297)
2024-03-12 17:46:19 +00:00
clap
Add tests for batching support (#29297)
2024-03-12 17:46:19 +00:00
clip
Adds pretrained IDs directly in the tests (#29534)
2024-03-13 14:53:27 +01:00
clipseg
Add tests for batching support (#29297)
2024-03-12 17:46:19 +00:00
clvp
Adds pretrained IDs directly in the tests (#29534)
2024-03-13 14:53:27 +01:00
code_llama
Adds pretrained IDs directly in the tests (#29534)
2024-03-13 14:53:27 +01:00
codegen
Adds pretrained IDs directly in the tests (#29534)
2024-03-13 14:53:27 +01:00
cohere
Cohere Model Release (#29622)
2024-03-15 14:29:11 +01:00
conditional_detr
Add tests for batching support (#29297)
2024-03-12 17:46:19 +00:00
convbert
Add tf_keras imports to prepare for Keras 3 (#28588)
2024-01-30 17:26:36 +00:00
convnext
Image Feature Extraction pipeline (#28216)
2024-02-05 14:50:07 +00:00
convnextv2
Image Feature Extraction pipeline (#28216)
2024-02-05 14:50:07 +00:00
cpm
Fix PipelineTests skip conditions (#22320)
2023-03-22 20:02:24 +01:00
cpmant
Adds pretrained IDs directly in the tests (#29534)
2024-03-13 14:53:27 +01:00
ctrl
Adds pretrained IDs directly in the tests (#29534)
2024-03-13 14:53:27 +01:00
cvt
Image Feature Extraction pipeline (#28216)
2024-02-05 14:50:07 +00:00
data2vec
Avoid using uncessary get_values(MODEL_MAPPING) (#29362)
2024-02-29 17:19:17 +08:00
deberta
Adds pretrained IDs directly in the tests (#29534)
2024-03-13 14:53:27 +01:00
deberta_v2
Adds pretrained IDs directly in the tests (#29534)
2024-03-13 14:53:27 +01:00
decision_transformer
🔥Rework pipeline testing by removing PipelineTestCaseMeta 🚀 (#21516)
2023-02-28 19:40:57 +01:00
deformable_detr
DeformableDETR support bfloat16 (#29232)
2024-03-04 14:18:09 +00:00
deit
Avoid using uncessary get_values(MODEL_MAPPING) (#29362)
2024-02-29 17:19:17 +08:00
depth_anything
Add Depth Anything (#28654)
2024-01-25 09:34:50 +01:00
deta
[YOLOS] Fix - return padded annotations (#29300)
2024-03-01 09:42:13 +00:00
detr
Add tests for batching support (#29297)
2024-03-12 17:46:19 +00:00
dinat
Image Feature Extraction pipeline (#28216)
2024-02-05 14:50:07 +00:00
dinov2
Image Feature Extraction pipeline (#28216)
2024-02-05 14:50:07 +00:00
distilbert
Adds pretrained IDs directly in the tests (#29534)
2024-03-13 14:53:27 +01:00
dit
Update old existing feature extractor references (#24552)
2023-06-29 10:17:36 +01:00
donut
Image Feature Extraction pipeline (#28216)
2024-02-05 14:50:07 +00:00
dpr
Adds pretrained IDs directly in the tests (#29534)
2024-03-13 14:53:27 +01:00
dpt
Add tests for batching support (#29297)
2024-03-12 17:46:19 +00:00
efficientformer
Avoid using uncessary get_values(MODEL_MAPPING) (#29362)
2024-02-29 17:19:17 +08:00
efficientnet
Image Feature Extraction pipeline (#28216)
2024-02-05 14:50:07 +00:00
electra
Adds pretrained IDs directly in the tests (#29534)
2024-03-13 14:53:27 +01:00
encodec
Add tests for batching support (#29297)
2024-03-12 17:46:19 +00:00
encoder_decoder
Generate: missing generation config eos token setting in encoder-decoder tests (#29146)
2024-02-20 16:17:51 +00:00
ernie
device agnostic models testing (#27146)
2023-10-31 18:12:14 +01:00
ernie_m
Adds pretrained IDs directly in the tests (#29534)
2024-03-13 14:53:27 +01:00
esm
FIX [quantization / ESM] Fix ESM 8bit / 4bit with bitsandbytes (#29329)
2024-03-01 03:01:53 +01:00
falcon
F.scaled_dot_product_attention support (#26572)
2023-12-09 05:38:14 +09:00
fastspeech2_conformer
Adds pretrained IDs directly in the tests (#29534)
2024-03-13 14:53:27 +01:00
flaubert
[tests] add the missing require_sacremoses decorator (#29504)
2024-03-08 10:13:54 +00:00
flava
Add tests for batching support (#29297)
2024-03-12 17:46:19 +00:00
fnet
Adds pretrained IDs directly in the tests (#29534)
2024-03-13 14:53:27 +01:00
focalnet
Image Feature Extraction pipeline (#28216)
2024-02-05 14:50:07 +00:00
fsmt
Adds pretrained IDs directly in the tests (#29534)
2024-03-13 14:53:27 +01:00
funnel
Adds pretrained IDs directly in the tests (#29534)
2024-03-13 14:53:27 +01:00
fuyu
Update tiny model summary file (#27388)
2023-11-23 21:00:39 +01:00
gemma
Adds pretrained IDs directly in the tests (#29534)
2024-03-13 14:53:27 +01:00
git
[core/ GC / tests] Stronger GC tests (#27124)
2023-10-30 19:53:46 +01:00
glpn
Avoid using uncessary get_values(MODEL_MAPPING) (#29362)
2024-02-29 17:19:17 +08:00
gpt2
Adds pretrained IDs directly in the tests (#29534)
2024-03-13 14:53:27 +01:00
gpt_bigcode
CI with num_hidden_layers=2 🚀🚀🚀 (#25266)
2023-08-02 20:22:36 +02:00
gpt_neo
Update all references to canonical models (#29001)
2024-02-16 08:16:58 +01:00
gpt_neox
[GPTNeoX] Fix BC issue with 4.36 (#28602)
2024-01-21 17:01:19 +00:00
gpt_neox_japanese
Adds pretrained IDs directly in the tests (#29534)
2024-03-13 14:53:27 +01:00
gpt_sw3
Adds pretrained IDs directly in the tests (#29534)
2024-03-13 14:53:27 +01:00
gptj
[Flash Attention 2] Add flash attention 2 for GPT-J (#28295)
2024-03-13 08:43:00 +01:00
gptsan_japanese
Adds pretrained IDs directly in the tests (#29534)
2024-03-13 14:53:27 +01:00
graphormer
Make more test models smaller (#25005)
2023-07-24 10:08:47 -04:00
groupvit
Add tests for batching support (#29297)
2024-03-12 17:46:19 +00:00
herbert
Adds pretrained IDs directly in the tests (#29534)
2024-03-13 14:53:27 +01:00
hubert
Fix failing tests on main due to torch 2.1 (#26607)
2023-10-05 10:27:05 +02:00
ibert
Update deprecated torch.range in test_modeling_ibert.py (#27355)
2023-11-08 20:58:36 +01:00
idefics
F.scaled_dot_product_attention support (#26572)
2023-12-09 05:38:14 +09:00
imagegpt
Image Feature Extraction pipeline (#28216)
2024-02-05 14:50:07 +00:00
informer
Add tests for batching support (#29297)
2024-03-12 17:46:19 +00:00
instructblip
Add tests for batching support (#29297)
2024-03-12 17:46:19 +00:00
jukebox
device agnostic models testing (#27146)
2023-10-31 18:12:14 +01:00
kosmos2
Add tests for batching support (#29297)
2024-03-12 17:46:19 +00:00
layoutlm
Adds pretrained IDs directly in the tests (#29534)
2024-03-13 14:53:27 +01:00
layoutlmv2
Adds pretrained IDs directly in the tests (#29534)
2024-03-13 14:53:27 +01:00
layoutlmv3
Adds pretrained IDs directly in the tests (#29534)
2024-03-13 14:53:27 +01:00
layoutxlm
Adds pretrained IDs directly in the tests (#29534)
2024-03-13 14:53:27 +01:00
led
Adds pretrained IDs directly in the tests (#29534)
2024-03-13 14:53:27 +01:00
levit
Avoid using uncessary get_values(MODEL_MAPPING) (#29362)
2024-02-29 17:19:17 +08:00
lilt
[core/ GC / tests] Stronger GC tests (#27124)
2023-10-30 19:53:46 +01:00
llama
Adds pretrained IDs directly in the tests (#29534)
2024-03-13 14:53:27 +01:00
llava
Fix _merge_input_ids_with_image_features for llava model (#28333)
2024-01-10 08:33:33 +01:00
longformer
Adds pretrained IDs directly in the tests (#29534)
2024-03-13 14:53:27 +01:00
longt5
Byebye torch 1.10 (#28207)
2024-01-11 16:18:27 +01:00
luke
Adds pretrained IDs directly in the tests (#29534)
2024-03-13 14:53:27 +01:00
lxmert
Adds pretrained IDs directly in the tests (#29534)
2024-03-13 14:53:27 +01:00
m2m_100
Adds pretrained IDs directly in the tests (#29534)
2024-03-13 14:53:27 +01:00
mamba
[Mamba doc] Post merge updates (#29472)
2024-03-11 09:46:24 +01:00
marian
Adds pretrained IDs directly in the tests (#29534)
2024-03-13 14:53:27 +01:00
markuplm
Adds pretrained IDs directly in the tests (#29534)
2024-03-13 14:53:27 +01:00
mask2former
Image Feature Extraction pipeline (#28216)
2024-02-05 14:50:07 +00:00
maskformer
Add tests for batching support (#29297)
2024-03-12 17:46:19 +00:00
mbart
Adds pretrained IDs directly in the tests (#29534)
2024-03-13 14:53:27 +01:00
mbart50
Adds pretrained IDs directly in the tests (#29534)
2024-03-13 14:53:27 +01:00
mega
fix / skip (for now) some tests before switch to torch 2.2 (#28838)
2024-02-02 14:11:50 +01:00
megatron_bert
CI with num_hidden_layers=2 🚀🚀🚀 (#25266)
2023-08-02 20:22:36 +02:00
megatron_gpt2
Move test model folders (#17034)
2022-05-03 14:42:02 +02:00
mgp_str
Adds pretrained IDs directly in the tests (#29534)
2024-03-13 14:53:27 +01:00
mistral
Flax mistral (#26943)
2024-01-31 14:19:02 +01:00
mixtral
Exclude the load balancing loss of padding tokens in Mixtral-8x7B (#28517)
2024-01-24 10:12:14 +01:00
mluke
Adds pretrained IDs directly in the tests (#29534)
2024-03-13 14:53:27 +01:00
mobilebert
Adds pretrained IDs directly in the tests (#29534)
2024-03-13 14:53:27 +01:00
mobilenet_v1
Image Feature Extraction pipeline (#28216)
2024-02-05 14:50:07 +00:00
mobilenet_v2
Add tests for batching support (#29297)
2024-03-12 17:46:19 +00:00
mobilevit
Image Feature Extraction pipeline (#28216)
2024-02-05 14:50:07 +00:00
mobilevitv2
Image Feature Extraction pipeline (#28216)
2024-02-05 14:50:07 +00:00
mpnet
Adds pretrained IDs directly in the tests (#29534)
2024-03-13 14:53:27 +01:00
mpt
Fix building alibi tensor when num_heads is not a power of 2 (#28380)
2024-01-08 10:39:40 +01:00
mra
Add tests for batching support (#29297)
2024-03-12 17:46:19 +00:00
mt5
Update all references to canonical models (#29001)
2024-02-16 08:16:58 +01:00
musicgen
Fix multi_gpu_data_parallel_forward for MusicgenTest (#29632)
2024-03-13 19:12:20 +01:00
musicgen_melody
Add MusicGen Melody (#28819)
2024-03-18 13:06:12 +00:00
mvp
Adds pretrained IDs directly in the tests (#29534)
2024-03-13 14:53:27 +01:00
nat
Image Feature Extraction pipeline (#28216)
2024-02-05 14:50:07 +00:00
nezha
CI with num_hidden_layers=2 🚀🚀🚀 (#25266)
2023-08-02 20:22:36 +02:00
nllb
Adds pretrained IDs directly in the tests (#29534)
2024-03-13 14:53:27 +01:00
nllb_moe
fix assisted decoding assistant model inputs (#27503)
2023-11-27 14:23:54 +00:00
nougat
Adds pretrained IDs directly in the tests (#29534)
2024-03-13 14:53:27 +01:00
nystromformer
CI with num_hidden_layers=2 🚀🚀🚀 (#25266)
2023-08-02 20:22:36 +02:00
oneformer
Fix OneFormer post_process_instance_segmentation for panoptic tasks (#29304)
2024-03-04 11:04:49 +00:00
openai
Adds pretrained IDs directly in the tests (#29534)
2024-03-13 14:53:27 +01:00
opt
Proper build() methods for TF (#27794)
2023-12-14 15:17:30 +00:00
owlv2
Add tests for batching support (#29297)
2024-03-12 17:46:19 +00:00
owlvit
Add tests for batching support (#29297)
2024-03-12 17:46:19 +00:00
patchtsmixer
PatchtTST and PatchTSMixer fixes (#28083)
2024-01-29 10:09:26 +00:00
patchtst
PatchtTST and PatchTSMixer fixes (#28083)
2024-01-29 10:09:26 +00:00
pegasus
Adds pretrained IDs directly in the tests (#29534)
2024-03-13 14:53:27 +01:00
pegasus_x
device agnostic models testing (#27146)
2023-10-31 18:12:14 +01:00
perceiver
Adds pretrained IDs directly in the tests (#29534)
2024-03-13 14:53:27 +01:00
persimmon
Fix device of masks in tests (#27887)
2023-12-07 21:34:43 +09:00
phi
[Phi] Extend implementation to use GQA/MQA. (#28163)
2024-01-11 15:58:02 +01:00
phobert
Adds pretrained IDs directly in the tests (#29534)
2024-03-13 14:53:27 +01:00
pix2struct
Add tests for batching support (#29297)
2024-03-12 17:46:19 +00:00
plbart
Adds pretrained IDs directly in the tests (#29534)
2024-03-13 14:53:27 +01:00
poolformer
Image Feature Extraction pipeline (#28216)
2024-02-05 14:50:07 +00:00
pop2piano
Byebye torch 1.10 (#28207)
2024-01-11 16:18:27 +01:00
prophetnet
Adds pretrained IDs directly in the tests (#29534)
2024-03-13 14:53:27 +01:00
pvt
Avoid using uncessary get_values(MODEL_MAPPING) (#29362)
2024-02-29 17:19:17 +08:00
pvt_v2
Fix PVT v2 tests (#29660)
2024-03-14 17:00:32 +01:00
qdqbert
Update all references to canonical models (#29001)
2024-02-16 08:16:58 +01:00
qwen2
Adds pretrained IDs directly in the tests (#29534)
2024-03-13 14:53:27 +01:00
rag
Add dataset_revision argument to RagConfig (#29610)
2024-03-14 16:48:11 +01:00
realm
Adds pretrained IDs directly in the tests (#29534)
2024-03-13 14:53:27 +01:00
reformer
Adds pretrained IDs directly in the tests (#29534)
2024-03-13 14:53:27 +01:00
regnet
Image Feature Extraction pipeline (#28216)
2024-02-05 14:50:07 +00:00
rembert
Adds pretrained IDs directly in the tests (#29534)
2024-03-13 14:53:27 +01:00
resnet
Image Feature Extraction pipeline (#28216)
2024-02-05 14:50:07 +00:00
roberta
Adds pretrained IDs directly in the tests (#29534)
2024-03-13 14:53:27 +01:00
roberta_prelayernorm
Update all references to canonical models (#29001)
2024-02-16 08:16:58 +01:00
roc_bert
Adds pretrained IDs directly in the tests (#29534)
2024-03-13 14:53:27 +01:00
roformer
Adds pretrained IDs directly in the tests (#29534)
2024-03-13 14:53:27 +01:00
rwkv
CI with num_hidden_layers=2 🚀🚀🚀 (#25266)
2023-08-02 20:22:36 +02:00
sam
Add tf_keras imports to prepare for Keras 3 (#28588)
2024-01-30 17:26:36 +00:00
seamless_m4t
Adds pretrained IDs directly in the tests (#29534)
2024-03-13 14:53:27 +01:00
seamless_m4t_v2
Fix languages covered by M4Tv2 (#28019)
2023-12-14 14:43:44 +00:00
segformer
Avoid using uncessary get_values(MODEL_MAPPING) (#29362)
2024-02-29 17:19:17 +08:00
seggpt
Fix batching tests for new models (Mamba and SegGPT) (#29633)
2024-03-13 17:52:49 +00:00
sew
Fix typo (#25966)
2023-09-05 10:12:25 +02:00
sew_d
Fix typo (#25966)
2023-09-05 10:12:25 +02:00
siglip
Adds pretrained IDs directly in the tests (#29534)
2024-03-13 14:53:27 +01:00
speech_encoder_decoder
Generate: missing generation config eos token setting in encoder-decoder tests (#29146)
2024-02-20 16:17:51 +00:00
speech_to_text
[FIX] Fix speech2test modeling tests (#29672)
2024-03-15 17:58:11 +00:00
speech_to_text_2
Adds pretrained IDs directly in the tests (#29534)
2024-03-13 14:53:27 +01:00
speecht5
Adds pretrained IDs directly in the tests (#29534)
2024-03-13 14:53:27 +01:00
splinter
CI with num_hidden_layers=2 🚀🚀🚀 (#25266)
2023-08-02 20:22:36 +02:00
squeezebert
Adds pretrained IDs directly in the tests (#29534)
2024-03-13 14:53:27 +01:00
stablelm
support SDPA Attention in stablelm (#29106)
2024-02-21 13:12:49 +01:00
starcoder2
FIX [CI / starcoder2] Change starcoder2 path to correct one for slow tests (#29359)
2024-02-29 03:52:13 +01:00
swiftformer
Image Feature Extraction pipeline (#28216)
2024-02-05 14:50:07 +00:00
swin
Image Feature Extraction pipeline (#28216)
2024-02-05 14:50:07 +00:00
swin2sr
Image Feature Extraction pipeline (#28216)
2024-02-05 14:50:07 +00:00
swinv2
Image Feature Extraction pipeline (#28216)
2024-02-05 14:50:07 +00:00
switch_transformers
Update all references to canonical models (#29001)
2024-02-16 08:16:58 +01:00
t5
Adds pretrained IDs directly in the tests (#29534)
2024-03-13 14:53:27 +01:00
table_transformer
Add tests for batching support (#29297)
2024-03-12 17:46:19 +00:00
tapas
Adds pretrained IDs directly in the tests (#29534)
2024-03-13 14:53:27 +01:00
time_series_transformer
Add tests for batching support (#29297)
2024-03-12 17:46:19 +00:00
timesformer
Update forward signature test for vision models (#27681)
2023-11-27 15:48:17 +01:00
timm_backbone
[AutoBackbone] Add test (#26094)
2023-09-18 23:47:54 +02:00
trocr
CI with num_hidden_layers=2 🚀🚀🚀 (#25266)
2023-08-02 20:22:36 +02:00
tvlt
remove failing tests and clean FE files (#27414)
2023-11-09 18:35:42 +00:00
tvp
Enable instantiating model with pretrained backbone weights (#28214)
2024-01-23 11:01:50 +00:00
udop
Adds pretrained IDs directly in the tests (#29534)
2024-03-13 14:53:27 +01:00
umt5
Update all references to canonical models (#29001)
2024-02-16 08:16:58 +01:00
unispeech
[Wav2Vec2 and Co] Update init tests for PT 2.1 (#26494)
2023-10-03 10:52:34 +02:00
unispeech_sat
Byebye torch 1.10 (#28207)
2024-01-11 16:18:27 +01:00
univnet
Add tests for batching support (#29297)
2024-03-12 17:46:19 +00:00
upernet
Enable instantiating model with pretrained backbone weights (#28214)
2024-01-23 11:01:50 +00:00
videomae
Update forward signature test for vision models (#27681)
2023-11-27 15:48:17 +01:00
vilt
Add tests for batching support (#29297)
2024-03-12 17:46:19 +00:00
vipllava
Fix _merge_input_ids_with_image_features for llava model (#28333)
2024-01-10 08:33:33 +01:00
vision_encoder_decoder
Fix VisionEncoderDecoder Positional Arg (#29497)
2024-03-07 20:45:51 +00:00
vision_text_dual_encoder
[Styling] stylify using ruff (#27144)
2023-11-16 17:43:19 +01:00
visual_bert
[core/ GC / tests] Stronger GC tests (#27124)
2023-10-30 19:53:46 +01:00
vit
Image Feature Extraction pipeline (#28216)
2024-02-05 14:50:07 +00:00
vit_hybrid
Add tests for batching support (#29297)
2024-03-12 17:46:19 +00:00
vit_mae
Add tests for batching support (#29297)
2024-03-12 17:46:19 +00:00
vit_msn
Image Feature Extraction pipeline (#28216)
2024-02-05 14:50:07 +00:00
vitdet
mark test_initialization as flaky in 2 model tests (#27906)
2023-12-08 14:54:32 +01:00
vitmatte
Enable instantiating model with pretrained backbone weights (#28214)
2024-01-23 11:01:50 +00:00
vits
Adds pretrained IDs directly in the tests (#29534)
2024-03-13 14:53:27 +01:00
vivit
Input data format (#25464)
2023-08-16 17:45:02 +01:00
wav2vec2
Adds pretrained IDs directly in the tests (#29534)
2024-03-13 14:53:27 +01:00
wav2vec2_bert
Add new meta w2v2-conformer BERT-like model (#28165)
2024-01-18 13:37:34 +00:00
wav2vec2_conformer
device agnostic models testing (#27146)
2023-10-31 18:12:14 +01:00
wav2vec2_phoneme
Adds pretrained IDs directly in the tests (#29534)
2024-03-13 14:53:27 +01:00
wav2vec2_with_lm
Fix some tests using "common_voice" (#27147)
2023-10-30 15:27:15 +01:00
wavlm
Byebye torch 1.10 (#28207)
2024-01-11 16:18:27 +01:00
whisper
Adds pretrained IDs directly in the tests (#29534)
2024-03-13 14:53:27 +01:00
x_clip
Add tests for batching support (#29297)
2024-03-12 17:46:19 +00:00
xglm
Adds pretrained IDs directly in the tests (#29534)
2024-03-13 14:53:27 +01:00
xlm
Adds pretrained IDs directly in the tests (#29534)
2024-03-13 14:53:27 +01:00
xlm_prophetnet
Adds pretrained IDs directly in the tests (#29534)
2024-03-13 14:53:27 +01:00
xlm_roberta
Adds pretrained IDs directly in the tests (#29534)
2024-03-13 14:53:27 +01:00
xlm_roberta_xl
Revert low cpu mem tie weights (#29135)
2024-02-20 12:06:46 +00:00
xlnet
Adds pretrained IDs directly in the tests (#29534)
2024-03-13 14:53:27 +01:00
xmod
Update all references to canonical models (#29001)
2024-02-16 08:16:58 +01:00
yolos
[YOLOS] Fix - return padded annotations (#29300)
2024-03-01 09:42:13 +00:00
yoso
CI with num_hidden_layers=2 🚀🚀🚀 (#25266)
2023-08-02 20:22:36 +02:00
__init__.py
Move test model folders (#17034)
2022-05-03 14:42:02 +02:00