Ryan Mullins 487dab1b2b Shieldgemma2 (#36678) ...

* single commit

* correct config

* fixup

* dummy pt

* Use ShieldGemma2Config in conversion script

* Update src/transformers/models/shieldgemma2/configuration_shieldgemma2.py

* Adding shieldgemma2 to models.__init__.py

* Adding ShieldGemma2 to main __init__.py

* Update shieldgemma2.md

* Update shieldgemma2.md

* Adding tests. Addressing review feedback.

* Minor docs update

* Fixing code quality feedback from CI

* Fixing empty messages bug reported by ghunkins

---------

Co-authored-by: Arthur Zucker <arthur.zucker@gmail.com>
Co-authored-by: Ren Pang <ain-soph@live.com>

2025-03-20 15:14:38 +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

Fix typos in tests (#36547)

2025-03-05 15:04:06 -08: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

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

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

Fix typos in tests (#36547)

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

blip_2

Fix typos in tests (#36547)

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

bloom

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

2025-02-17 14:59:22 +00: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

Fix typos in tests (#36547)

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

chameleon

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

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

chinese_clip

Fix typos in tests (#36547)

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

clap

Fix typos in tests (#36547)

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

clip

Fix typos in tests (#36547)

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

clipseg

Fix typos in tests (#36547)

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

clvp

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

2025-02-21 10:18:20 +00: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

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

2025-02-21 10:18:20 +00: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

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

2025-02-21 10:18:20 +00: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

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

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

deit

Fix typos in tests (#36547)

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

depth_anything

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

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

depth_pro

Fix typos in tests (#36547)

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

detr

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

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

diffllama

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

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

Fix typos in tests (#36547)

2025-03-05 15:04:06 -08: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

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

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

encoder_decoder

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

2025-02-19 15:13:27 +00: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

[tests] enable bnb tests on xpu (#36233)

2025-02-24 11:30:15 +01:00

falcon_mamba

[docs] Redesign (#31757)

2025-03-03 10:33:46 -08:00

fastspeech2_conformer

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

2025-01-24 16:55:28 +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

Fix typos in tests (#36547)

2025-03-05 15:04:06 -08: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

Gemma3 (#36658)

2025-03-12 09:06:17 +01:00

gemma2

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

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

gemma3

[gemma 3] multimodal checkpoints + AutoModelForCausalLM (#36741)

2025-03-19 15:04:19 +00: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

fix tiktoken convert to pass AddedToken to Tokenizer (#36566)

2025-03-20 11:26:49 +01:00

gpt_bigcode

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

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

gpt_neo

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

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

gpt_neox

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

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

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

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

granitemoe

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

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

Fix typos in tests (#36547)

2025-03-05 15:04:06 -08: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

Fix typos in tests (#36547)

2025-03-05 15:04:06 -08: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

[tests] remove pt_tf equivalence tests (#36253)

2025-02-19 11:55:11 +00: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

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

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

llava

Refactor some core stuff (#36539)

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

llava_next

Fix typos in tests (#36547)

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

llava_next_video

fix xpu tests (#36656)

2025-03-17 15:57:49 +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

Fix typos in tests (#36547)

2025-03-05 15:04:06 -08: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

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

2025-01-24 16:55:28 +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

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

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

mllama

Fix bugs in mllama image processing (#36156)

2025-03-11 10:22:48 +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

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

2025-02-11 11:37:31 +00: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

[tests] Parameterized test_eager_matches_sdpa_inference (#36650)

2025-03-14 14:41:27 +00: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

[tests] Parameterized test_eager_matches_sdpa_inference (#36650)

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

musicgen_melody

[tests] Parameterized test_eager_matches_sdpa_inference (#36650)

2025-03-14 14:41:27 +00: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

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

2025-01-24 16:55:28 +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

Fix typos in tests (#36547)

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

owlvit

Fix typos in tests (#36547)

2025-03-05 15:04:06 -08: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

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

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

phi

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

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

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

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

2025-03-18 18:39:36 +00: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

Fix typos in tests (#36547)

2025-03-05 15:04:06 -08: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

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

2025-02-11 11:37:31 +00: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

Fix typos in tests (#36547)

2025-03-05 15:04:06 -08: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

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

2025-02-19 15:13:27 +00: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

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

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

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

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

table_transformer

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

2025-02-21 10:18:20 +00: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

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

2025-01-29 16:05:33 +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

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

2025-01-24 16:55:28 +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

[tests] Parameterized test_eager_matches_sdpa_inference (#36650)

2025-03-14 14:41:27 +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

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

2025-02-19 15:13:27 +00: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

[tests] remove pt_tf equivalence tests (#36253)

2025-02-19 11:55:11 +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

Fix typos in tests (#36547)

2025-03-05 15:04:06 -08: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

Fix typos in tests (#36547)

2025-03-05 15:04:06 -08: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

Fix typos in tests (#36547)

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

xglm

[XGLM] tag tests as slow (#36592)

2025-03-06 17:54:41 +00: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

…