riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-31 02:02:21 +06:00
Code Issues Actions 33 Packages Projects Releases Wiki Activity
0e4a1c3401
transformers/tests/models
History
Saurabh Dash 0e4a1c3401
Cohere Model Release (#29622) 
* Cohere Model Release (#1)

Cohere Model Release

* Remove unnecessary files and code (#2)

Some cleanup

* Delete cohere-model directory (#3)

* Make Fix (#5)

* Pr fixes (#6)

* fixes for pr

* pr fixes for the format

* pr fixes for the format

* src/transformers/models/auto/tokenization_auto.py

* Tokenizer test (#8)

* tokenizer test

* format fix

* Adding Docs and other minor changes (#7)

* Add modeling tests (#9)

* Smol Fix (#11)

* tokenization tests are fixed

* format fixes

* fix pr doc tests

* fix pr doc tests

* fix pr doc tests

* fix pr style check

* small changes in cohere.md

* FIX: Address final comments for transformers integration (#13)

* fix modeling final nits and add proper test file

* for now leave empty tests

* add integration test

* push new test

* fix modeling cohere (#14)

* Update chat templates to use the new API (#15)

---------

Co-authored-by: ahmetustun <ahmetustun89@gmail.com>
Co-authored-by: Younes Belkada <49240599+younesbelkada@users.noreply.github.com>
Co-authored-by: Matt <Rocketknight1@users.noreply.github.com>
2024-03-15 14:29:11 +01: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
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
gpt2 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
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 Adds pretrained IDs directly in the tests (#29534) 2024-03-13 14:53:27 +01: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