amyeroberts bdf36dcd48 Enable HF pretrained backbones (#31145) ...

* Enable load HF or tim backbone checkpoints

* Fix up

* Fix test - pass in proper out_indices

* Update docs

* Fix tvp tests

* Fix doc examples

* Fix doc examples

* Try to resolve DPT backbone param init

* Don't conditionally set to None

* Add condition based on whether backbone is defined

* Address review comments

2024-06-06 22:02:38 +01:00

..

albert

Remove static pretrained maps from the library's internals (#29112)

2024-03-25 10:33:38 +01:00

align

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

altclip

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

audio_spectrogram_transformer

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

auto

Fix auto tests (#30067)

2024-04-05 17:49:46 +02:00

autoformer

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

bark

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

bart

update ruff version (#30932)

2024-05-22 06:40:15 +02: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

Enable dynamic resolution input for Beit (#31053)

2024-06-06 14:47:41 +01:00

bert

Ignore non-causal mask in more cases with SDPA (#30138)

2024-06-03 19:08:41 +08: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

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

bigbird_pegasus

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

biogpt

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

bit

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

blenderbot

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

blenderbot_small

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

blip

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

blip_2

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

bloom

Remove static pretrained maps from the library's internals (#29112)

2024-03-25 10:33:38 +01:00

bridgetower

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

bros

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

byt5

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

camembert

Adds pretrained IDs directly in the tests (#29534)

2024-03-13 14:53:27 +01:00

canine

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

chinese_clip

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

clap

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

clip

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

clipseg

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

clvp

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

code_llama

Adds pretrained IDs directly in the tests (#29534)

2024-03-13 14:53:27 +01:00

codegen

Add token type ids to CodeGenTokenizer (#29265)

2024-04-17 12:19:18 +02:00

cohere

Fix failing tokenizer tests (#31083)

2024-05-28 13:34:23 +02:00

conditional_detr

Enable HF pretrained backbones (#31145)

2024-06-06 22:02:38 +01:00

convbert

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

convnext

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

convnextv2

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

cpm

Fix PipelineTests skip conditions (#22320)

2023-03-22 20:02:24 +01:00

cpmant

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

ctrl

Remove static pretrained maps from the library's internals (#29112)

2024-03-25 10:33:38 +01:00

cvt

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

data2vec

Enable dynamic resolution input for Beit (#31053)

2024-06-06 14:47:41 +01:00

dbrx

Fix accelerate failing tests (#30836)

2024-05-23 17:18:58 +02:00

deberta

Remove static pretrained maps from the library's internals (#29112)

2024-03-25 10:33:38 +01:00

deberta_v2

Remove static pretrained maps from the library's internals (#29112)

2024-03-25 10:33:38 +01:00

decision_transformer

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

deformable_detr

Enable HF pretrained backbones (#31145)

2024-06-06 22:02:38 +01:00

deit

Add dynamic resolution input/interpolate position embedding to deit (#31131)

2024-06-04 10:29:01 +01:00

depth_anything

Enable HF pretrained backbones (#31145)

2024-06-06 22:02:38 +01:00

detr

Enable HF pretrained backbones (#31145)

2024-06-06 22:02:38 +01:00

dinat

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

dinov2

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

distilbert

Fix FA2 tests (#29909)

2024-04-01 07:51:00 +00:00

dit

Update old existing feature extractor references (#24552)

2023-06-29 10:17:36 +01:00

donut

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

dpr

Remove static pretrained maps from the library's internals (#29112)

2024-03-25 10:33:38 +01:00

dpt

Enable HF pretrained backbones (#31145)

2024-06-06 22:02:38 +01:00

efficientnet

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

electra

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

encodec

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

encoder_decoder

[tests] add torch.use_deterministic_algorithms for XPU (#30774)

2024-05-23 16:53:07 +01:00

ernie

Remove static pretrained maps from the library's internals (#29112)

2024-03-25 10:33:38 +01:00

esm

Fix sentence fragment within test comments (#31218)

2024-06-04 10:09:24 +01:00

falcon

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

fastspeech2_conformer

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

flaubert

Add missing Flaubert tokenizer tests (#30492)

2024-06-05 13:52:16 +02:00

flava

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

fnet

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

focalnet

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

fsmt

Encoder-decoder models: move embedding scale to nn.Module (#30410)

2024-05-01 12:33:00 +05:00

funnel

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

fuyu

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

gemma

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

git

fix: Replace deprecated assertEquals with assertEqual (#30241)

2024-04-15 09:36:06 +01:00

glpn

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

gpt2

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

gpt_bigcode

CI with num_hidden_layers=2 🚀🚀🚀 (#25266)

2023-08-02 20:22:36 +02:00

gpt_neo

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

gpt_neox

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

gpt_neox_japanese

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

gpt_sw3

Adds pretrained IDs directly in the tests (#29534)

2024-03-13 14:53:27 +01:00

gptj

Remove static pretrained maps from the library's internals (#29112)

2024-03-25 10:33:38 +01:00

grounding_dino

Enable HF pretrained backbones (#31145)

2024-06-06 22:02:38 +01:00

groupvit

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

herbert

Adds pretrained IDs directly in the tests (#29534)

2024-03-13 14:53:27 +01:00

hubert

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

ibert

Fix sentence fragment within test comments (#31218)

2024-06-04 10:09:24 +01:00

idefics

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

idefics2

Encoder-decoder models: move embedding scale to nn.Module (#30410)

2024-05-01 12:33:00 +05:00

imagegpt

Encoder-decoder models: move embedding scale to nn.Module (#30410)

2024-05-01 12:33:00 +05:00

informer

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

instructblip

TST: Fix instruct-blip tests (#31088)

2024-05-28 18:29:11 +02:00

jamba

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

jetmoe

Skip failing JetMOE generation tests (#31266)

2024-06-05 19:06:46 +01:00

kosmos2

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

layoutlm

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

layoutlmv2

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

layoutlmv3

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

layoutxlm

Add split special tokens (#30772)

2024-05-24 08:38:58 -07:00

led

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

levit

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

lilt

Remove static pretrained maps from the library's internals (#29112)

2024-03-25 10:33:38 +01:00

llama

Fix test_compile_static_cache (#30991)

2024-06-03 15:16:28 +02:00

llava

Support arbitrary processor (#30875)

2024-05-17 16:51:31 +02:00

llava_next

LLaVa-Next: Update docs with batched inference (#30857)

2024-05-20 13:45:56 +05:00

longformer

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

longt5

Do not remove half seq length in generation tests (#30016)

2024-04-19 17:32:52 +01:00

luke

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

lxmert

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

m2m_100

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

mamba

Make mamba use cache (#31116)

2024-06-06 13:37:29 +05:00

marian

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

markuplm

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

mask2former

Enable HF pretrained backbones (#31145)

2024-06-06 22:02:38 +01:00

maskformer

Enable HF pretrained backbones (#31145)

2024-06-06 22:02:38 +01:00

mbart

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

mbart50

Adds pretrained IDs directly in the tests (#29534)

2024-03-13 14:53:27 +01:00

megatron_bert

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

megatron_gpt2

…

mgp_str

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

mistral

Fix MistralIntegrationTest (#31231)

2024-06-04 18:04:08 +02:00

mixtral

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

mluke

Adds pretrained IDs directly in the tests (#29534)

2024-03-13 14:53:27 +01:00

mobilebert

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

mobilenet_v1

Mark MobileNetV1ModelTest::test_batching_equivalence as flaky (#31258)

2024-06-06 14:47:58 +01:00

mobilenet_v2

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

mobilevit

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

mobilevitv2

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

mpnet

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

mpt

Update 4 MptIntegrationTests expected outputs (#30989)

2024-05-23 18:27:54 +02:00

mra

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

mt5

Fix accelerate failing tests (#30836)

2024-05-23 17:18:58 +02:00

musicgen

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

musicgen_melody

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

mvp

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

nllb

Remove deprecated properties in tokenization_nllb.py and tokenization_nllb_fast.py (#29834)

2024-05-23 18:53:26 +02:00

nllb_moe

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

nougat

Adds pretrained IDs directly in the tests (#29534)

2024-03-13 14:53:27 +01:00

nystromformer

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

olmo

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

oneformer

Enable HF pretrained backbones (#31145)

2024-06-06 22:02:38 +01:00

openai

Remove static pretrained maps from the library's internals (#29112)

2024-03-25 10:33:38 +01:00

opt

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

owlv2

Fix OWLv2 post_process_object_detection for multiple images (#31082)

2024-05-28 12:06:06 +01:00

owlvit

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

paligemma

Paligemma causal attention mask (#30967)

2024-05-22 19:37:15 +02:00

patchtsmixer

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

patchtst

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

pegasus

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

pegasus_x

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

perceiver

Perceiver interpolate position embedding (#30979)

2024-05-24 11:13:58 +01:00

persimmon

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

phi

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

phi3

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

phobert

Adds pretrained IDs directly in the tests (#29534)

2024-03-13 14:53:27 +01:00

pix2struct

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

plbart

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

poolformer

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

pop2piano

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

prophetnet

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

pvt

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

pvt_v2

Remove static pretrained maps from the library's internals (#29112)

2024-03-25 10:33:38 +01:00

qwen2

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

qwen2_moe

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

rag

Add dataset_revision argument to RagConfig (#29610)

2024-03-14 16:48:11 +01:00

recurrent_gemma

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

reformer

Do not remove half seq length in generation tests (#30016)

2024-04-19 17:32:52 +01:00

regnet

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

rembert

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

resnet

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

roberta

Fix sentence fragment within test comments (#31218)

2024-06-04 10:09:24 +01:00

roberta_prelayernorm

Fix sentence fragment within test comments (#31218)

2024-06-04 10:09:24 +01:00

roc_bert

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

roformer

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

rwkv

Remove ninja from docker image build (#31080)

2024-05-28 11:36:26 +02:00

sam

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

seamless_m4t

Remove deprecated properties in tokenization_nllb.py and tokenization_nllb_fast.py (#29834)

2024-05-23 18:53:26 +02:00

seamless_m4t_v2

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

segformer

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

seggpt

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

sew

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

sew_d

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

siglip

Fix accelerate failing tests (#30836)

2024-05-23 17:18:58 +02:00

speech_encoder_decoder

Deprecate low use models (#30781)

2024-05-28 18:07:07 +01:00

speech_to_text

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

speecht5

[tests] add torch.use_deterministic_algorithms for XPU (#30774)

2024-05-23 16:53:07 +01:00

splinter

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

squeezebert

Remove static pretrained maps from the library's internals (#29112)

2024-03-25 10:33:38 +01:00

stablelm

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

starcoder2

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

superpoint

Removal of deprecated maps (#30576)

2024-05-09 14:15:56 +02:00

swiftformer

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

swin

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

swin2sr

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

swinv2

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

switch_transformers

Fix accelerate failing tests (#30836)

2024-05-23 17:18:58 +02:00

t5

Fix accelerate failing tests (#30836)

2024-05-23 17:18:58 +02:00

table_transformer

Enable HF pretrained backbones (#31145)

2024-06-06 22:02:38 +01:00

tapas

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

time_series_transformer

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

timesformer

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

timm_backbone

🚨 out_indices always a list (#30941)

2024-05-22 15:23:04 +01:00

trocr

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

tvp

Enable HF pretrained backbones (#31145)

2024-06-06 22:02:38 +01:00

udop

Add split special tokens (#30772)

2024-05-24 08:38:58 -07:00

umt5

Fix accelerate failing tests (#30836)

2024-05-23 17:18:58 +02:00

unispeech

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

unispeech_sat

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

univnet

Add tests for batching support (#29297)

2024-03-12 17:46:19 +00:00

upernet

Enable HF pretrained backbones (#31145)

2024-06-06 22:02:38 +01:00

video_llava

Video-LLaVa: handle any number of frames (#31221)

2024-06-04 14:20:03 +05:00

videomae

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

vilt

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

vipllava

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

vision_encoder_decoder

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

vision_text_dual_encoder

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

visual_bert

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

vit

skip test_multi_gpu_data_parallel_forward for vit and deit (#31086)

2024-05-28 17:44:52 +02:00

vit_mae

added interpolation for vitmae model in pytorch as well as tf. (#30732)

2024-05-24 16:20:09 +01:00

vit_msn

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

vitdet

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

vitmatte

Enable HF pretrained backbones (#31145)

2024-06-06 22:02:38 +01:00

vits

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

vivit

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

wav2vec2

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

wav2vec2_bert

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

wav2vec2_conformer

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

wav2vec2_phoneme

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

wav2vec2_with_lm

Add AutoFeatureExtractor support to Wav2Vec2ProcessorWithLM (#28706)

2024-05-20 13:40:42 +02:00

wavlm

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

whisper

Fix pad_to_max_length Whisper (#30787)

2024-05-27 16:09:05 +02:00

x_clip

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

xglm

Remove static pretrained maps from the library's internals (#29112)

2024-03-25 10:33:38 +01:00

xlm

Fix 29807 sinusoidal positional encodings in Flaubert, Informer and XLM (#29904)

2024-04-02 10:27:26 +02:00

xlm_roberta

Adds pretrained IDs directly in the tests (#29534)

2024-03-13 14:53:27 +01:00

xlm_roberta_xl

fix accelerate tests for roberta xl (#31288)

2024-06-06 14:44:35 +01:00

xlnet

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

xmod

Fix sentence fragment within test comments (#31218)

2024-06-04 10:09:24 +01:00

yolos

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

yoso

update ruff version (#30932)

2024-05-22 06:40:15 +02:00

__init__.py

…