Fanli Lin 4dbf17c17f [tests] enable bnb tests on xpu (#36233) ...

* fix failed test

* fix device

* fix more device cases

* add more cases

* fix empty cache

* Update test_4bit.py

---------

Co-authored-by: Yih-Dar <2521628+ydshieh@users.noreply.github.com>

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

..

albert

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

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

align

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

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

altclip

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

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

aria

Add @require_bitsandbytes to Aria test_batched_generation (#36192)

2025-02-14 15:48:47 +01: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

[tests] remove pt_tf equivalence tests (#36253)

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

autoformer

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

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

bamba

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

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

bark

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

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

bart

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

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

Fix tests for vision models (#35654)

2025-02-13 10:28:37 +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

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

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

bigbird_pegasus

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

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

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

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

blip_2

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

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

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

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

chameleon

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

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

chinese_clip

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

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

clap

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

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

clip

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

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

clipseg

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

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

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

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

colpali

Add support for nested images to LLava and VipLLava (#35558)

2025-01-30 16:49:20 -05: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] remove flax-pt equivalence and cross tests (#36283)

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

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

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

Move DataCollatorForMultipleChoice from the docs to the package (#34763)

2025-02-13 12:01:28 +01: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 tests for vision models (#35654)

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

dinov2_with_registers

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

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

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

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

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

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

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

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

fnet

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

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

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

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

gemma

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

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

gemma2

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

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

git

GitModelIntegrationTest - flatten the expected slice tensor (#36260)

2025-02-18 16:04:19 +01: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

🔴 VLM: compile compatibility (#35724)

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

gpt2

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

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

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

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

groupvit

[tests] remove pt_tf equivalence tests (#36253)

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

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

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

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

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

informer

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

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

instructblip

🔴 VLM: compile compatibility (#35724)

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

instructblipvideo

🔴 VLM: compile compatibility (#35724)

2025-02-14 15:23:49 +01: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

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

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

layoutlm

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

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

layoutlmv2

[tests] remove pt_tf equivalence tests (#36253)

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

layoutlmv3

[tests] remove pt_tf equivalence tests (#36253)

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

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

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

llama

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

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

llava

Fix LlavaForConditionalGenerationModelTest::test_config after #36077 (#36230)

2025-02-17 11:49:07 +01:00

llava_next

🔴 VLM: compile compatibility (#35724)

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

llava_next_video

Uniformize LlavaNextVideoProcessor kwargs (#35613)

2025-02-18 14:13:51 -05:00

llava_onevision

🔴 VLM: compile compatibility (#35724)

2025-02-14 15:23:49 +01: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

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

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

luke

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

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

lxmert

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

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

m2m_100

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

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

mamba

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

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

mamba2

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

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

marian

Fix broken CI on release branch due to missing conversion files (#36275)

2025-02-20 13:22:10 +01: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

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

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

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

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

mistral

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

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

mixtral

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

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

mllama

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

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

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

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

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

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

mt5

[tests] remove test_export_to_onnx (#36241)

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

musicgen

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

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

musicgen_melody

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

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

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

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

Uniformize OwlViT and Owlv2 processors (#35700)

2025-02-13 17:30:26 -05:00

owlvit

Uniformize OwlViT and Owlv2 processors (#35700)

2025-02-13 17:30:26 -05:00

paligemma

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

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

paligemma2

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

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

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

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

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

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

pixtral

Refactoring of ImageProcessorFast (#35069)

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

plbart

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

2025-02-13 16:27:11 +01: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] remove cache v4.47 deprecations (#36212)

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

qwen2_audio

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

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

qwen2_moe

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

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

qwen2_vl

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

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

rag

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

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

recurrent_gemma

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

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

reformer

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

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

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

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

[tests] remove pt_tf equivalence tests (#36253)

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

seamless_m4t

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

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

seamless_m4t_v2

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

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

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

SmolVLM2 (#36126)

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

speech_encoder_decoder

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

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

speech_to_text

[tests] deflake dither test (#36284)

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

speecht5

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

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

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

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

swinv2

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

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

switch_transformers

[tests] remove test_export_to_onnx (#36241)

2025-02-17 16:52:44 +00: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

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

2025-02-17 14:57:50 +01: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

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

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

umt5

[tests] remove test_export_to_onnx (#36241)

2025-02-17 16:52:44 +00: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

🔴 VLM: compile compatibility (#35724)

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

videomae

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

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

vilt

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

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

vipllava

🔴 VLM: compile compatibility (#35724)

2025-02-14 15:23:49 +01: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

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

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

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

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

vitdet

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

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

vitmatte

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

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

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

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

wav2vec2_bert

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

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

wav2vec2_conformer

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

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

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

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

x_clip

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

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

xglm

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

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

Move test model folders (#17034)

2022-05-03 14:42:02 +02:00