transformers/tests/models
jiqing-feng 0e56fb69a2
fix pegasus init weights and other copied models (#36844)
* fix pegasus init weights

Signed-off-by: jiqing-feng <jiqing.feng@intel.com>

* fix the rest of models

Signed-off-by: jiqing-feng <jiqing.feng@intel.com>

* fix test

Signed-off-by: jiqing-feng <jiqing.feng@intel.com>

* fix informer init

Signed-off-by: jiqing-feng <jiqing.feng@intel.com>

* init weight before checking

Signed-off-by: jiqing-feng <jiqing.feng@intel.com>

* fix roformer tests

Signed-off-by: jiqing-feng <jiqing.feng@intel.com>

* fix roformer tests

Signed-off-by: jiqing-feng <jiqing.feng@intel.com>

---------

Signed-off-by: jiqing-feng <jiqing.feng@intel.com>
2025-03-27 14:14:30 +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 Support return_tensors in audio chat templates (#34601) 2025-03-25 11:08: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 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 fix typos in the tests directory (#36932) 2025-03-25 10:49:24 +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 fix typos in the tests directory (#36932) 2025-03-25 10:49:24 +01: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 Update ruff to 0.11.2 (#36962) 2025-03-25 16:00:11 +01: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 Deprecate #36741 and map Causal to Conditional (#36917) 2025-03-25 09:13:56 +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
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 Update ruff to 0.11.2 (#36962) 2025-03-25 16:00: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
gpt2 Update ruff to 0.11.2 (#36962) 2025-03-25 16:00:11 +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 fix typos in the tests directory (#36932) 2025-03-25 10:49:24 +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 fix typos in the tests directory (#36932) 2025-03-25 10:49:24 +01: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 pegasus init weights and other copied models (#36844) 2025-03-27 14:14:30 +01: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 Fix typos (#36910) 2025-03-24 14:08:29 +00: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 Update ruff to 0.11.2 (#36962) 2025-03-25 16:00:11 +01:00
maskformer Update ruff to 0.11.2 (#36962) 2025-03-25 16:00:11 +01: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 fix typos in the tests directory (#36932) 2025-03-25 10:49:24 +01:00
mistral [tests] remove tf/flax tests in /generation (#36235) 2025-02-17 14:59:22 +00:00
mistral3 Deprecate #36741 and map Causal to Conditional (#36917) 2025-03-25 09:13:56 +01:00
mixtral Fix tensor dtype mismatch (#36985) 2025-03-26 10:37:46 +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 Support QuestionAnswering Module for ModernBert based models. (#35566) 2025-03-26 21:24:18 +01: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 (#36910) 2025-03-24 14:08:29 +00:00
paligemma2 Fix typos (#36910) 2025-03-24 14:08:29 +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 pegasus init weights and other copied models (#36844) 2025-03-27 14:14:30 +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 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 Export for Phi4-mini (#36780) 2025-03-25 10:46:38 +01:00
phi4_multimodal Add Phi4 multimodal (#36939) 2025-03-25 09:55:21 +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 Support return_tensors in audio chat templates (#34601) 2025-03-25 11:08:47 +01:00
qwen2_audio Support return_tensors in audio chat templates (#34601) 2025-03-25 11:08:47 +01:00
qwen2_moe Update configuration_qwen2.py (#36735) 2025-03-19 18:15:54 +00:00
qwen2_vl Support return_tensors in audio chat templates (#34601) 2025-03-25 11:08:47 +01: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 pegasus init weights and other copied models (#36844) 2025-03-27 14:14:30 +01:00
rt_detr fix typos in the tests directory (#36932) 2025-03-25 10:49:24 +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 Support return_tensors in audio chat templates (#34601) 2025-03-25 11:08:47 +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 Mark 2 tests as flaky for now (#37038) 2025-03-27 10:59:47 +01: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 the tests directory (#36932) 2025-03-25 10:49:24 +01: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 (#36910) 2025-03-24 14:08:29 +00: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 Support return_tensors in audio chat templates (#34601) 2025-03-25 11:08:47 +01:00
wav2vec2_bert Support return_tensors in audio chat templates (#34601) 2025-03-25 11:08:47 +01: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 Mark 2 tests as flaky for now (#37038) 2025-03-27 10:59:47 +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