Raushan Turganbay b469ebc5cf Prepend bos token to Blip generations (#29642) ...

* prepend "bos" to blip generation

* minor changes

* Update src/transformers/models/blip_2/modeling_blip_2.py

Co-authored-by: Joao Gante <joaofranciscocardosogante@gmail.com>

* Update src/transformers/models/instructblip/modeling_instructblip.py

Co-authored-by: amyeroberts <22614925+amyeroberts@users.noreply.github.com>

* add generation tester mixin

---------

Co-authored-by: Joao Gante <joaofranciscocardosogante@gmail.com>
Co-authored-by: amyeroberts <22614925+amyeroberts@users.noreply.github.com>

2024-03-21 16:33:18 +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

Prepend bos token to Blip generations (#29642)

2024-03-21 16:33:18 +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

[BC 4.37 -> 4.38] for Llama family, memory and speed (#29753)

2024-03-20 23:47:01 +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

Prepend bos token to Blip generations (#29642)

2024-03-21 16:33:18 +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

Add correct batched handling for apply_chat_template (#29222)

2024-03-20 15:50:22 +00:00

layoutlmv3

Add correct batched handling for apply_chat_template (#29222)

2024-03-20 15:50:22 +00:00

layoutxlm

Add correct batched handling for apply_chat_template (#29222)

2024-03-20 15:50:22 +00: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

[BC 4.37 -> 4.38] for Llama family, memory and speed (#29753)

2024-03-20 23:47:01 +01:00

llava

Add LLaVa-1.6, bis (#29586)

2024-03-20 15:51:12 +00:00

llava_next

Add LLaVa-1.6, bis (#29586)

2024-03-20 15:51:12 +00: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

Add correct batched handling for apply_chat_template (#29222)

2024-03-20 15:50:22 +00: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

Tests: Musicgen tests + make fix-copies (#29734)

2024-03-20 08:45:53 +01:00

musicgen_melody

Tests: Musicgen tests + make fix-copies (#29734)

2024-03-20 08:45:53 +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

[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

superpoint

SuperPointModel -> SuperPointForKeypointDetection (#29757)

2024-03-20 15:41:03 +00: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

Add correct batched handling for apply_chat_template (#29222)

2024-03-20 15:50:22 +00: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

Add correct batched handling for apply_chat_template (#29222)

2024-03-20 15:50:22 +00: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

Add LLaVa-1.6, bis (#29586)

2024-03-20 15:51:12 +00: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