Cyril Vallez dd3933dd65 Simplify keep_in_fp32_modules logic (#36722) ...

* better regex everywhere

* fix

* Update test_modeling_instructblip.py

* BC with explanations this time otherwise it makes no sense at all

* Update test_modeling_instructblip.py

* style

* CIs

* update _keep_in_fp32_modules in blip2

* Update modeling_utils.py

* Update modeling_utils.py

* style

* CIs

* add check

* trigger CIs

* Update modeling_utils.py

* trigger CIs

2025-03-21 16:12:59 +01:00

..

albert

[tests] Parameterized test_eager_matches_sdpa_inference (#36650)

2025-03-14 14:41:27 +00:00

align

chore: fix typos in tests directory (#36785)

2025-03-18 10:31:13 +01:00

altclip

chore: fix typos in tests directory (#36785)

2025-03-18 10:31:13 +01:00

aria

Fix typos in tests (#36547)

2025-03-05 15:04:06 -08:00

audio_spectrogram_transformer

use torch.testing.assertclose instead to get more details about error in cis (#35659)

2025-01-24 16:55:28 +01:00

auto

chore: fix typos in tests directory (#36785)

2025-03-18 10:31:13 +01:00

autoformer

Fix typos in tests (#36547)

2025-03-05 15:04:06 -08:00

aya_vision

[Generation, Gemma 3] When passing a custom generation_config, overwrite default values with the model's base generation_config (#36684)

2025-03-15 12:40:09 +00:00

bamba

Expectations test utils (#36569)

2025-03-18 23:39:50 +01:00

bark

fix can_generate (#36570)

2025-03-17 14:56:18 +01:00

bart

chore: fix typos in the tests directory (#36813)

2025-03-21 10:20:05 +01:00

barthez

Skip tests properly (#31308)

2024-06-26 21:59:08 +01:00

bartpho

Adds pretrained IDs directly in the tests (#29534)

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

beit

[tests] Parameterized test_eager_matches_sdpa_inference (#36650)

2025-03-14 14:41:27 +00:00

bert

chore: fix typos in the tests directory (#36813)

2025-03-21 10:20:05 +01:00

bert_generation

CI: avoid human error, automatically infer generative models (#33212)

2025-02-13 16:27:11 +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

Fix typos in tests (#36547)

2025-03-05 15:04:06 -08:00

bigbird_pegasus

Fix typos in tests (#36547)

2025-03-05 15:04:06 -08:00

biogpt

CI: avoid human error, automatically infer generative models (#33212)

2025-02-13 16:27:11 +01:00

bit

Add common test for torch.export and fix some vision models (#35124)

2025-02-11 11:37:31 +00:00

blenderbot

[tests] remove tf/flax tests in /generation (#36235)

2025-02-17 14:59:22 +00:00

blenderbot_small

[tests] remove tf/flax tests in /generation (#36235)

2025-02-17 14:59:22 +00:00

blip

chore: fix typos in the tests directory (#36813)

2025-03-21 10:20:05 +01:00

blip_2

chore: fix typos in the tests directory (#36813)

2025-03-21 10:20:05 +01:00

bloom

chore: fix typos in the tests directory (#36813)

2025-03-21 10:20:05 +01:00

bridgetower

[tests] further fix Tester object has no attribute '_testMethodName' (#35781)

2025-01-29 16:05:33 +01:00

bros

CI: avoid human error, automatically infer generative models (#33212)

2025-02-13 16:27:11 +01:00

byt5

Skip tests properly (#31308)

2024-06-26 21:59:08 +01:00

camembert

use torch.testing.assertclose instead to get more details about error in cis (#35659)

2025-01-24 16:55:28 +01:00

canine

chore: fix typos in the tests directory (#36813)

2025-03-21 10:20:05 +01:00

chameleon

CI: avoid human error, automatically infer generative models (#33212)

2025-02-13 16:27:11 +01:00

chinese_clip

chore: fix typos in the tests directory (#36813)

2025-03-21 10:20:05 +01:00

clap

chore: fix typos in the tests directory (#36813)

2025-03-21 10:20:05 +01:00

clip

chore: fix typos in the tests directory (#36813)

2025-03-21 10:20:05 +01:00

clipseg

chore: fix typos in the tests directory (#36813)

2025-03-21 10:20:05 +01:00

clvp

chore: fix typos in the tests directory (#36813)

2025-03-21 10:20:05 +01:00

code_llama

Skip tests properly (#31308)

2024-06-26 21:59:08 +01:00

codegen

CI: avoid human error, automatically infer generative models (#33212)

2025-02-13 16:27:11 +01:00

cohere

CI: avoid human error, automatically infer generative models (#33212)

2025-02-13 16:27:11 +01:00

cohere2

[generate] ✨ vectorized beam search ✨ (#35802)

2025-03-18 18:39:36 +00:00

colpali

[tests] Parameterized test_eager_matches_sdpa_inference (#36650)

2025-03-14 14:41:27 +00:00

conditional_detr

chore: fix typos in the tests directory (#36813)

2025-03-21 10:20:05 +01:00

convbert

use torch.testing.assertclose instead to get more details about error in cis (#35659)

2025-01-24 16:55:28 +01:00

convnext

Add common test for torch.export and fix some vision models (#35124)

2025-02-11 11:37:31 +00:00

convnextv2

Add common test for torch.export and fix some vision models (#35124)

2025-02-11 11:37:31 +00:00

cpm

Make pipeline able to load processor (#32514)

2024-10-09 16:46:11 +01:00

cpmant

chore: fix typos in the tests directory (#36813)

2025-03-21 10:20:05 +01:00

ctrl

CI: avoid human error, automatically infer generative models (#33212)

2025-02-13 16:27:11 +01:00

cvt

Add common test for torch.export and fix some vision models (#35124)

2025-02-11 11:37:31 +00:00

dab_detr

[CI] Check test if the GenerationTesterMixin inheritance is correct 🐛 🔫 (#36180)

2025-02-21 10:18:20 +00:00

dac

use torch.testing.assertclose instead to get more details about error in cis (#35659)

2025-01-24 16:55:28 +01:00

data2vec

[tests] Parameterized test_eager_matches_sdpa_inference (#36650)

2025-03-14 14:41:27 +00:00

dbrx

CI: avoid human error, automatically infer generative models (#33212)

2025-02-13 16:27:11 +01:00

deberta

[tests] remove pt_tf equivalence tests (#36253)

2025-02-19 11:55:11 +00:00

deberta_v2

[tests] remove pt_tf equivalence tests (#36253)

2025-02-19 11:55:11 +00:00

decision_transformer

[CI] Check test if the GenerationTesterMixin inheritance is correct 🐛 🔫 (#36180)

2025-02-21 10:18:20 +00:00

deformable_detr

chore: fix typos in the tests directory (#36813)

2025-03-21 10:20:05 +01:00

deit

Fix typos in tests (#36547)

2025-03-05 15:04:06 -08:00

depth_anything

chore: fix typos in the tests directory (#36813)

2025-03-21 10:20:05 +01:00

depth_pro

chore: fix typos in the tests directory (#36813)

2025-03-21 10:20:05 +01:00

detr

chore: fix typos in the tests directory (#36813)

2025-03-21 10:20:05 +01:00

diffllama

chore: fix typos in the tests directory (#36813)

2025-03-21 10:20:05 +01:00

dinat

Add common test for torch.export and fix some vision models (#35124)

2025-02-11 11:37:31 +00:00

dinov2

Fix typos in tests (#36547)

2025-03-05 15:04:06 -08:00

dinov2_with_registers

Fix typos in tests (#36547)

2025-03-05 15:04:06 -08:00

distilbert

[tests] further fix Tester object has no attribute '_testMethodName' (#35781)

2025-01-29 16:05:33 +01:00

dit

use torch.testing.assertclose instead to get more details about error in cis (#35659)

2025-01-24 16:55:28 +01:00

donut

[tests] further fix Tester object has no attribute '_testMethodName' (#35781)

2025-01-29 16:05:33 +01:00

dpr

use torch.testing.assertclose instead to get more details about error in cis (#35659)

2025-01-24 16:55:28 +01:00

dpt

chore: fix typos in the tests directory (#36813)

2025-03-21 10:20:05 +01:00

efficientnet

Add common test for torch.export and fix some vision models (#35124)

2025-02-11 11:37:31 +00:00

electra

[CI] Check test if the GenerationTesterMixin inheritance is correct 🐛 🔫 (#36180)

2025-02-21 10:18:20 +00:00

emu3

🔴 VLM: compile compatibility (#35724)

2025-02-14 15:23:49 +01:00

encodec

chore: fix typos in the tests directory (#36813)

2025-03-21 10:20:05 +01:00

encoder_decoder

chore: fix typos in the tests directory (#36813)

2025-03-21 10:20:05 +01:00

ernie

CI: avoid human error, automatically infer generative models (#33212)

2025-02-13 16:27:11 +01:00

esm

[tests] fix EsmModelIntegrationTest::test_inference_bitsandbytes (#36225)

2025-02-17 11:10:33 +01:00

falcon

chore: fix typos in the tests directory (#36813)

2025-03-21 10:20:05 +01:00

falcon_mamba

chore: fix typos in the tests directory (#36813)

2025-03-21 10:20:05 +01:00

fastspeech2_conformer

chore: fix typos in the tests directory (#36813)

2025-03-21 10:20:05 +01:00

flaubert

[CI] Check test if the GenerationTesterMixin inheritance is correct 🐛 🔫 (#36180)

2025-02-21 10:18:20 +00:00

flava

Fix typos in tests (#36547)

2025-03-05 15:04:06 -08:00

fnet

chore: fix typos in the tests directory (#36813)

2025-03-21 10:20:05 +01:00

focalnet

Add common test for torch.export and fix some vision models (#35124)

2025-02-11 11:37:31 +00:00

fsmt

[tests] remove test_export_to_onnx (#36241)

2025-02-17 16:52:44 +00:00

funnel

use torch.testing.assertclose instead to get more details about error in cis (#35659)

2025-01-24 16:55:28 +01:00

fuyu

[Generation, Gemma 3] When passing a custom generation_config, overwrite default values with the model's base generation_config (#36684)

2025-03-15 12:40:09 +00:00

gemma

chore: fix typos in the tests directory (#36813)

2025-03-21 10:20:05 +01:00

gemma2

[generate] ✨ vectorized beam search ✨ (#35802)

2025-03-18 18:39:36 +00:00

gemma3

Gemma 3 tests expect greedy decoding (#36882)

2025-03-21 12:36:39 +01:00

git

Fix typos in tests (#36547)

2025-03-05 15:04:06 -08:00

glm

CI: avoid human error, automatically infer generative models (#33212)

2025-02-13 16:27:11 +01:00

glpn

Add common test for torch.export and fix some vision models (#35124)

2025-02-11 11:37:31 +00:00

got_ocr2

Add Got-OCR 2 Fast image processor and refactor slow one (#36185)

2025-03-01 00:56:00 -05:00

gpt2

chore: fix typos in the tests directory (#36813)

2025-03-21 10:20:05 +01:00

gpt_bigcode

CI: avoid human error, automatically infer generative models (#33212)

2025-02-13 16:27:11 +01:00

gpt_neo

chore: fix typos in the tests directory (#36813)

2025-03-21 10:20:05 +01:00

gpt_neox

chore: fix typos in the tests directory (#36813)

2025-03-21 10:20:05 +01:00

gpt_neox_japanese

CI: avoid human error, automatically infer generative models (#33212)

2025-02-13 16:27:11 +01:00

gpt_sw3

🚨 No more default chat templates (#31733)

2024-07-24 17:36:32 +01:00

gptj

[tests] remove flax-pt equivalence and cross tests (#36283)

2025-02-19 15:13:27 +00:00

granite

chore: fix typos in the tests directory (#36813)

2025-03-21 10:20:05 +01:00

granitemoe

chore: fix typos in the tests directory (#36813)

2025-03-21 10:20:05 +01:00

granitemoeshared

add shared experts for upcoming Granite 4.0 language models (#35894)

2025-02-14 16:55:28 +01:00

grounding_dino

[GroundingDino] Fix grounding dino loss 🚨 (#31828)

2025-02-27 19:15:58 +00:00

groupvit

chore: fix typos in the tests directory (#36813)

2025-03-21 10:20:05 +01:00

helium

CI: avoid human error, automatically infer generative models (#33212)

2025-02-13 16:27:11 +01:00

herbert

Skip tests properly (#31308)

2024-06-26 21:59:08 +01:00

hiera

Add common test for torch.export and fix some vision models (#35124)

2025-02-11 11:37:31 +00:00

hubert

[tests] remove pt_tf equivalence tests (#36253)

2025-02-19 11:55:11 +00:00

ibert

[warnings] fix E721 warnings (#32223)

2024-07-25 15:12:23 +02:00

idefics

[tests] Parameterized test_eager_matches_sdpa_inference (#36650)

2025-03-14 14:41:27 +00:00

idefics2

CI: avoid human error, automatically infer generative models (#33212)

2025-02-13 16:27:11 +01:00

idefics3

SmolVLM2 (#36126)

2025-02-20 15:00:26 +01:00

ijepa

Add common test for torch.export and fix some vision models (#35124)

2025-02-11 11:37:31 +00:00

imagegpt

Fix typos in tests (#36547)

2025-03-05 15:04:06 -08:00

informer

Fix typos in tests (#36547)

2025-03-05 15:04:06 -08:00

instructblip

Simplify keep_in_fp32_modules logic (#36722)

2025-03-21 16:12:59 +01:00

instructblipvideo

Fix typos in tests (#36547)

2025-03-05 15:04:06 -08:00

jamba

CI: avoid human error, automatically infer generative models (#33212)

2025-02-13 16:27:11 +01:00

jetmoe

CI: avoid human error, automatically infer generative models (#33212)

2025-02-13 16:27:11 +01:00

kosmos2

Remove hardcoded slow image processor class in processors supporting fast ones (#36266)

2025-03-12 18:39:25 -04:00

layoutlm

Fix typos in tests (#36547)

2025-03-05 15:04:06 -08:00

layoutlmv2

chore: fix typos in the tests directory (#36813)

2025-03-21 10:20:05 +01:00

layoutlmv3

Fix typos in tests (#36547)

2025-03-05 15:04:06 -08:00

layoutxlm

[tests] remove pt_tf equivalence tests (#36253)

2025-02-19 11:55:11 +00:00

led

CI: avoid human error, automatically infer generative models (#33212)

2025-02-13 16:27:11 +01:00

levit

[tests] further fix Tester object has no attribute '_testMethodName' (#35781)

2025-01-29 16:05:33 +01:00

lilt

Fix typos in tests (#36547)

2025-03-05 15:04:06 -08:00

llama

chore: fix typos in the tests directory (#36813)

2025-03-21 10:20:05 +01:00

llava

Refactor some core stuff (#36539)

2025-03-11 09:26:28 +01:00

llava_next

chore: fix typos in the tests directory (#36813)

2025-03-21 10:20:05 +01:00

llava_next_video

chore: fix typos in the tests directory (#36813)

2025-03-21 10:20:05 +01:00

llava_onevision

Fix typos in tests (#36547)

2025-03-05 15:04:06 -08:00

longformer

use torch.testing.assertclose instead to get more details about error in cis (#35659)

2025-01-24 16:55:28 +01:00

longt5

Fix typos in tests (#36547)

2025-03-05 15:04:06 -08:00

luke

Fix typos in tests (#36547)

2025-03-05 15:04:06 -08:00

lxmert

Fix typos in tests (#36547)

2025-03-05 15:04:06 -08:00

m2m_100

chore: fix typos in the tests directory (#36813)

2025-03-21 10:20:05 +01:00

mamba

Fix typos in tests (#36547)

2025-03-05 15:04:06 -08:00

mamba2

CI: avoid human error, automatically infer generative models (#33212)

2025-02-13 16:27:11 +01:00

marian

Fix typos in tests (#36547)

2025-03-05 15:04:06 -08:00

markuplm

[tests] remove pt_tf equivalence tests (#36253)

2025-02-19 11:55:11 +00:00

mask2former

Add common test for torch.export and fix some vision models (#35124)

2025-02-11 11:37:31 +00:00

maskformer

Add common test for torch.export and fix some vision models (#35124)

2025-02-11 11:37:31 +00:00

mbart

Fix typos in tests (#36547)

2025-03-05 15:04:06 -08:00

mbart50

Skip tests properly (#31308)

2024-06-26 21:59:08 +01:00

megatron_bert

[CI] Check test if the GenerationTesterMixin inheritance is correct 🐛 🔫 (#36180)

2025-02-21 10:18:20 +00:00

megatron_gpt2

Skip tests properly (#31308)

2024-06-26 21:59:08 +01:00

mgp_str

chore: fix typos in the tests directory (#36813)

2025-03-21 10:20:05 +01:00

mimi

[tests] Parameterized test_eager_matches_sdpa_inference (#36650)

2025-03-14 14:41:27 +00:00

mistral

[tests] remove tf/flax tests in /generation (#36235)

2025-02-17 14:59:22 +00:00

mistral3

Fix Mistral3 tests (#36797)

2025-03-18 13:08:12 -04:00

mixtral

chore: fix typos in the tests directory (#36813)

2025-03-21 10:20:05 +01:00

mllama

fix: loss computation after embeddings resize - mllama (#36840)

2025-03-21 14:47:59 +01:00

mluke

fix: Updated test_embeded_special_tokens for luke and mluke models (#32413)

2024-08-05 15:19:42 +01:00

mobilebert

MobileBERT is ExecuTorch compatible (#34473)

2024-10-29 16:14:31 +01:00

mobilenet_v1

Add common test for torch.export and fix some vision models (#35124)

2025-02-11 11:37:31 +00:00

mobilenet_v2

Add common test for torch.export and fix some vision models (#35124)

2025-02-11 11:37:31 +00:00

mobilevit

chore: fix typos in the tests directory (#36813)

2025-03-21 10:20:05 +01:00

mobilevitv2

Add common test for torch.export and fix some vision models (#35124)

2025-02-11 11:37:31 +00:00

modernbert

[CI] Check test if the GenerationTesterMixin inheritance is correct 🐛 🔫 (#36180)

2025-02-21 10:18:20 +00:00

moonshine

[CI] Check test if the GenerationTesterMixin inheritance is correct 🐛 🔫 (#36180)

2025-02-21 10:18:20 +00:00

moshi

chore: fix typos in the tests directory (#36813)

2025-03-21 10:20:05 +01:00

mpnet

use torch.testing.assertclose instead to get more details about error in cis (#35659)

2025-01-24 16:55:28 +01:00

mpt

CI: avoid human error, automatically infer generative models (#33212)

2025-02-13 16:27:11 +01:00

mra

Fix typos in tests (#36547)

2025-03-05 15:04:06 -08:00

mt5

[tests] remove test_export_to_onnx (#36241)

2025-02-17 16:52:44 +00:00

musicgen

chore: fix typos in the tests directory (#36813)

2025-03-21 10:20:05 +01:00

musicgen_melody

chore: fix typos in the tests directory (#36813)

2025-03-21 10:20:05 +01:00

mvp

CI: avoid human error, automatically infer generative models (#33212)

2025-02-13 16:27:11 +01:00

myt5

[Red CIs] Fix hub failures (#34001)

2024-10-07 10:56:24 +02:00

nemotron

CI: avoid human error, automatically infer generative models (#33212)

2025-02-13 16:27:11 +01:00

nllb

Skip tests properly (#31308)

2024-06-26 21:59:08 +01:00

nllb_moe

Fix typos in tests (#36547)

2025-03-05 15:04:06 -08:00

nougat

[tests] further fix Tester object has no attribute '_testMethodName' (#35781)

2025-01-29 16:05:33 +01:00

nystromformer

use torch.testing.assertclose instead to get more details about error in cis (#35659)

2025-01-24 16:55:28 +01:00

olmo

CI: avoid human error, automatically infer generative models (#33212)

2025-02-13 16:27:11 +01:00

olmo2

CI: avoid human error, automatically infer generative models (#33212)

2025-02-13 16:27:11 +01:00

olmoe

CI: avoid human error, automatically infer generative models (#33212)

2025-02-13 16:27:11 +01:00

omdet_turbo

use torch.testing.assertclose instead to get more details about error in cis (#35659)

2025-01-24 16:55:28 +01:00

oneformer

chore: fix typos in the tests directory (#36813)

2025-03-21 10:20:05 +01:00

openai

CI: avoid human error, automatically infer generative models (#33212)

2025-02-13 16:27:11 +01:00

opt

[tests] remove tf/flax tests in /generation (#36235)

2025-02-17 14:59:22 +00:00

owlv2

chore: fix typos in the tests directory (#36813)

2025-03-21 10:20:05 +01:00

owlvit

chore: fix typos in the tests directory (#36813)

2025-03-21 10:20:05 +01:00

paligemma

Fix typos in tests (#36547)

2025-03-05 15:04:06 -08:00

paligemma2

[Generation, Gemma 3] When passing a custom generation_config, overwrite default values with the model's base generation_config (#36684)

2025-03-15 12:40:09 +00:00

patchtsmixer

CI: avoid human error, automatically infer generative models (#33212)

2025-02-13 16:27:11 +01:00

patchtst

Add missing atol to torch.testing.assert_close where rtol is specified (#36234)

2025-02-17 14:57:50 +01:00

pegasus

Fix typos in tests (#36547)

2025-03-05 15:04:06 -08:00

pegasus_x

[CI] Check test if the GenerationTesterMixin inheritance is correct 🐛 🔫 (#36180)

2025-02-21 10:18:20 +00:00

perceiver

use torch.testing.assertclose instead to get more details about error in cis (#35659)

2025-01-24 16:55:28 +01:00

persimmon

chore: fix typos in the tests directory (#36813)

2025-03-21 10:20:05 +01:00

phi

chore: fix typos in the tests directory (#36813)

2025-03-21 10:20:05 +01:00

phi3

CI: avoid human error, automatically infer generative models (#33212)

2025-02-13 16:27:11 +01:00

phimoe

[generate] remove cache v4.47 deprecations (#36212)

2025-02-17 13:55:03 +00:00

phobert

Adds pretrained IDs directly in the tests (#29534)

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

pix2struct

Fix typos in tests (#36547)

2025-03-05 15:04:06 -08:00

pixtral

Add Mistral3 (#36790)

2025-03-18 12:04:42 +01:00

plbart

Fix typos in tests (#36547)

2025-03-05 15:04:06 -08:00

poolformer

Add common test for torch.export and fix some vision models (#35124)

2025-02-11 11:37:31 +00:00

pop2piano

[CI] Check test if the GenerationTesterMixin inheritance is correct 🐛 🔫 (#36180)

2025-02-21 10:18:20 +00:00

prompt_depth_anything

Add Prompt Depth Anything Model (#35401)

2025-03-20 16:12:44 +00:00

prophetnet

CI: avoid human error, automatically infer generative models (#33212)

2025-02-13 16:27:11 +01:00

pvt

Add common test for torch.export and fix some vision models (#35124)

2025-02-11 11:37:31 +00:00

pvt_v2

Add common test for torch.export and fix some vision models (#35124)

2025-02-11 11:37:31 +00:00

qwen2

Revert qwen2 breaking changes related to attention refactor (#36162)

2025-02-14 13:44:14 +01:00

qwen2_5_vl

chore: fix typos in the tests directory (#36813)

2025-03-21 10:20:05 +01:00

qwen2_audio

[CI] Check test if the GenerationTesterMixin inheritance is correct 🐛 🔫 (#36180)

2025-02-21 10:18:20 +00:00

qwen2_moe

Update configuration_qwen2.py (#36735)

2025-03-19 18:15:54 +00:00

qwen2_vl

[generate] ✨ vectorized beam search ✨ (#35802)

2025-03-18 18:39:36 +00:00

rag

[tests] make cuda-only tests device-agnostic (#35222)

2024-12-18 10:14:22 +01:00

recurrent_gemma

[Generation, Gemma 3] When passing a custom generation_config, overwrite default values with the model's base generation_config (#36684)

2025-03-15 12:40:09 +00:00

reformer

chore: fix typos in the tests directory (#36813)

2025-03-21 10:20:05 +01:00

regnet

Add common test for torch.export and fix some vision models (#35124)

2025-02-11 11:37:31 +00:00

rembert

[CI] Check test if the GenerationTesterMixin inheritance is correct 🐛 🔫 (#36180)

2025-02-21 10:18:20 +00:00

resnet

Add common test for torch.export and fix some vision models (#35124)

2025-02-11 11:37:31 +00:00

roberta

CI: avoid human error, automatically infer generative models (#33212)

2025-02-13 16:27:11 +01:00

roberta_prelayernorm

CI: avoid human error, automatically infer generative models (#33212)

2025-02-13 16:27:11 +01:00

roc_bert

[CI] Check test if the GenerationTesterMixin inheritance is correct 🐛 🔫 (#36180)

2025-02-21 10:18:20 +00:00

roformer

Fix typos in tests (#36547)

2025-03-05 15:04:06 -08:00

rt_detr

Au revoir flaky test_fast_is_faster_than_slow (#36240)

2025-02-17 18:30:07 +01:00

rt_detr_v2

chore: fix typos in the tests directory (#36813)

2025-03-21 10:20:05 +01:00

rwkv

CI: avoid human error, automatically infer generative models (#33212)

2025-02-13 16:27:11 +01:00

sam

🚨🚨🚨 Fix sdpa in SAM and refactor relative position embeddings (#36422)

2025-03-17 09:39:52 +00:00

seamless_m4t

chore: fix typos in the tests directory (#36813)

2025-03-21 10:20:05 +01:00

seamless_m4t_v2

Fix typos in tests (#36547)

2025-03-05 15:04:06 -08:00

segformer

[tests] remove pt_tf equivalence tests (#36253)

2025-02-19 11:55:11 +00:00

seggpt

Add common test for torch.export and fix some vision models (#35124)

2025-02-11 11:37:31 +00:00

sew

use torch.testing.assertclose instead to get more details about error in cis (#35659)

2025-01-24 16:55:28 +01:00

sew_d

use torch.testing.assertclose instead to get more details about error in cis (#35659)

2025-01-24 16:55:28 +01:00

shieldgemma2

Shieldgemma2 (#36678)

2025-03-20 15:14:38 +01:00

siglip

Refactoring of ImageProcessorFast (#35069)

2025-02-04 17:52:31 -05:00

siglip2

Add SigLIP 2 (#36323)

2025-02-21 09:04:19 +00:00

smolvlm

[Generation, Gemma 3] When passing a custom generation_config, overwrite default values with the model's base generation_config (#36684)

2025-03-15 12:40:09 +00:00

speech_encoder_decoder

chore: fix typos in the tests directory (#36813)

2025-03-21 10:20:05 +01:00

speech_to_text

Fix typos in tests (#36547)

2025-03-05 15:04:06 -08:00

speecht5

fix can_generate (#36570)

2025-03-17 14:56:18 +01:00

splinter

Make pipeline able to load processor (#32514)

2024-10-09 16:46:11 +01:00

squeezebert

use torch.testing.assertclose instead to get more details about error in cis (#35659)

2025-01-24 16:55:28 +01:00

stablelm

chore: fix typos in the tests directory (#36813)

2025-03-21 10:20:05 +01:00

starcoder2

CI: avoid human error, automatically infer generative models (#33212)

2025-02-13 16:27:11 +01:00

superglue

CI: avoid human error, automatically infer generative models (#33212)

2025-02-13 16:27:11 +01:00

superpoint

CI: avoid human error, automatically infer generative models (#33212)

2025-02-13 16:27:11 +01:00

swiftformer

Add common test for torch.export and fix some vision models (#35124)

2025-02-11 11:37:31 +00:00

swin

Add common test for torch.export and fix some vision models (#35124)

2025-02-11 11:37:31 +00:00

swin2sr

Fix typos in tests (#36547)

2025-03-05 15:04:06 -08:00

swinv2

Add common test for torch.export and fix some vision models (#35124)

2025-02-11 11:37:31 +00:00

switch_transformers

Fix typos in tests (#36547)

2025-03-05 15:04:06 -08:00

t5

chore: fix typos in the tests directory (#36813)

2025-03-21 10:20:05 +01:00

table_transformer

chore: fix typos in the tests directory (#36813)

2025-03-21 10:20:05 +01:00

tapas

[tests] remove pt_tf equivalence tests (#36253)

2025-02-19 11:55:11 +00:00

textnet

Fix tests for vision models (#35654)

2025-02-13 10:28:37 +00:00

time_series_transformer

Fix typos in tests (#36547)

2025-03-05 15:04:06 -08:00

timesformer

Add common test for torch.export and fix some vision models (#35124)

2025-02-11 11:37:31 +00:00

timm_backbone

Less flaky for TimmBackboneModelTest::test_batching_equivalence (#35971)

2025-01-30 16:56:26 +01:00

timm_wrapper

Enable auto task for timm models in pipeline (#35531)

2025-01-08 15:14:17 +00:00

trocr

CI: avoid human error, automatically infer generative models (#33212)

2025-02-13 16:27:11 +01:00

tvp

chore: fix typos in the tests directory (#36813)

2025-03-21 10:20:05 +01:00

udop

Fix typos in tests (#36547)

2025-03-05 15:04:06 -08:00

umt5

Fix typos in tests (#36547)

2025-03-05 15:04:06 -08:00

unispeech

use torch.testing.assertclose instead to get more details about error in cis (#35659)

2025-01-24 16:55:28 +01:00

unispeech_sat

use torch.testing.assertclose instead to get more details about error in cis (#35659)

2025-01-24 16:55:28 +01:00

univnet

chore: fix typos in the tests directory (#36813)

2025-03-21 10:20:05 +01:00

upernet

Add common test for torch.export and fix some vision models (#35124)

2025-02-11 11:37:31 +00:00

video_llava

Fix typos in tests (#36547)

2025-03-05 15:04:06 -08:00

videomae

Refactor Attention implementation for ViT-based models (#36545)

2025-03-20 15:15:01 +00:00

vilt

Fix typos in tests (#36547)

2025-03-05 15:04:06 -08:00

vipllava

Fix typos in tests (#36547)

2025-03-05 15:04:06 -08:00

vision_encoder_decoder

chore: fix typos in the tests directory (#36813)

2025-03-21 10:20:05 +01:00

vision_text_dual_encoder

[tests] remove flax-pt equivalence and cross tests (#36283)

2025-02-19 15:13:27 +00:00

visual_bert

Fix typos in tests (#36547)

2025-03-05 15:04:06 -08:00

vit

Fix tests for vision models (#35654)

2025-02-13 10:28:37 +00:00

vit_mae

Refactor Attention implementation for ViT-based models (#36545)

2025-03-20 15:15:01 +00:00

vit_msn

[tests] Parameterized test_eager_matches_sdpa_inference (#36650)

2025-03-14 14:41:27 +00:00

vitdet

Fixed VitDet for non-squre Images (#35969)

2025-02-25 19:31:24 +00:00

vitmatte

chore: fix typos in the tests directory (#36813)

2025-03-21 10:20:05 +01:00

vitpose

Add common test for torch.export and fix some vision models (#35124)

2025-02-11 11:37:31 +00:00

vitpose_backbone

skip test_initialization for VitPoseBackboneModelTest for now (#36154)

2025-02-12 18:24:24 +01:00

vits

use torch.testing.assertclose instead to get more details about error in cis (#35659)

2025-01-24 16:55:28 +01:00

vivit

Add common test for torch.export and fix some vision models (#35124)

2025-02-11 11:37:31 +00:00

wav2vec2

chore: fix typos in the tests directory (#36813)

2025-03-21 10:20:05 +01:00

wav2vec2_bert

Fix typos in tests (#36547)

2025-03-05 15:04:06 -08:00

wav2vec2_conformer

Fix typos in tests (#36547)

2025-03-05 15:04:06 -08:00

wav2vec2_phoneme

clean_up_tokenization_spaces=False if unset (#31938)

2024-09-26 19:38:20 +02:00

wav2vec2_with_lm

use torch.testing.assertclose instead to get more details about error in cis (#35659)

2025-01-24 16:55:28 +01:00

wavlm

Add missing atol to torch.testing.assert_close where rtol is specified (#36234)

2025-02-17 14:57:50 +01:00

whisper

Fix typos in tests (#36547)

2025-03-05 15:04:06 -08:00

x_clip

chore: fix typos in the tests directory (#36813)

2025-03-21 10:20:05 +01:00

xglm

chore: fix typos in the tests directory (#36813)

2025-03-21 10:20:05 +01:00

xlm

CI: avoid human error, automatically infer generative models (#33212)

2025-02-13 16:27:11 +01:00

xlm_roberta

use torch.testing.assertclose instead to get more details about error in cis (#35659)

2025-01-24 16:55:28 +01:00

xlm_roberta_xl

CI: avoid human error, automatically infer generative models (#33212)

2025-02-13 16:27:11 +01:00

xlnet

CI: avoid human error, automatically infer generative models (#33212)

2025-02-13 16:27:11 +01:00

xmod

CI: avoid human error, automatically infer generative models (#33212)

2025-02-13 16:27:11 +01:00

yolos

Add common test for torch.export and fix some vision models (#35124)

2025-02-11 11:37:31 +00:00

yoso

CI: avoid human error, automatically infer generative models (#33212)

2025-02-13 16:27:11 +01:00

zamba

CI: avoid human error, automatically infer generative models (#33212)

2025-02-13 16:27:11 +01:00

zamba2

CI: avoid human error, automatically infer generative models (#33212)

2025-02-13 16:27:11 +01:00

zoedepth

Fix tests for vision models (#35654)

2025-02-13 10:28:37 +00:00

__init__.py

…