transformers/tests/models
Karel Vesely 1a81d774b1
Add dithering to the Speech2TextFeatureExtractor API. (#34638)
* Add dithering to the `Speech2TextFeatureExtractor` API.

- in kaldi : 4a8b7f6732/src/feat/feature-window.cc (L145)
- with dithering without a seed, the features become non-deterministic due
  to small Gaussian noise added to the audio (i.e. 2 runs lead to little
  different outputs)

* update the PR

- add dithering also for WhisperFeatureExtractor
- not adding to Wav2Vec2FeatureExtractor (no FBANK computation)

* add unit-tests for dithering, fix docstrings

* ruff

* utils/check_copies.py --fix_and_overwrite

* update code, add seed to unit-test

* adding explanation of dithering
2025-02-19 11:50:02 +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 Fall back to slow image processor in ImageProcessingAuto when no fast processor available (#34785) 2024-12-15 14:00:36 -05: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: avoid human error, automatically infer generative models (#33212) 2025-02-13 16:27:11 +01: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 Refactoring of ImageProcessorFast (#35069) 2025-02-04 17:52:31 -05:00
blip_2 set test_torchscript = False for Blip2 testing (#35972) 2025-02-14 17:43:32 +01: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 Refactoring of ImageProcessorFast (#35069) 2025-02-04 17:52:31 -05:00
clipseg use torch.testing.assertclose instead to get more details about error in cis (#35659) 2025-01-24 16:55:28 +01:00
clvp CI: avoid human error, automatically infer generative models (#33212) 2025-02-13 16:27:11 +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 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 Add common test for torch.export and fix some vision models (#35124) 2025-02-11 11:37:31 +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 use torch.testing.assertclose instead to get more details about error in cis (#35659) 2025-01-24 16:55:28 +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 Add common test for torch.export and fix some vision models (#35124) 2025-02-11 11:37:31 +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 CI: avoid human error, automatically infer generative models (#33212) 2025-02-13 16:27:11 +01:00
dbrx CI: avoid human error, automatically infer generative models (#33212) 2025-02-13 16:27:11 +01:00
deberta use torch.testing.assertclose instead to get more details about error in cis (#35659) 2025-01-24 16:55:28 +01:00
deberta_v2 use torch.testing.assertclose instead to get more details about error in cis (#35659) 2025-01-24 16:55:28 +01:00
decision_transformer CI: avoid human error, automatically infer generative models (#33212) 2025-02-13 16:27:11 +01:00
deformable_detr Add common test for torch.export and fix some vision models (#35124) 2025-02-11 11:37:31 +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 Fix tests for vision models (#35654) 2025-02-13 10:28:37 +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 [tests] further fix Tester object has no attribute '_testMethodName' (#35781) 2025-01-29 16:05:33 +01: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 🚨All attention refactor🚨 (#35235) 2024-12-18 16:53:39 +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 CI: avoid human error, automatically infer generative models (#33212) 2025-02-13 16:27:11 +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 use torch.testing.assertclose instead to get more details about error in cis (#35659) 2025-01-24 16:55:28 +01: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
gpt_bigcode CI: avoid human error, automatically infer generative models (#33212) 2025-02-13 16:27:11 +01:00
gpt_neo [tests] remove tf/flax tests in /generation (#36235) 2025-02-17 14:59:22 +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
gpt2 [tests] remove tf/flax tests in /generation (#36235) 2025-02-17 14:59:22 +00:00
gptj [tests] remove tf/flax tests in /generation (#36235) 2025-02-17 14:59:22 +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 use torch.testing.assertclose instead to get more details about error in cis (#35659) 2025-01-24 16:55:28 +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 use torch.testing.assertclose instead to get more details about error in cis (#35659) 2025-01-24 16:55:28 +01:00
ibert [warnings] fix E721 warnings (#32223) 2024-07-25 15:12:23 +02:00
idefics 🔴 VLM: compile compatibility (#35724) 2025-02-14 15:23:49 +01:00
idefics2 CI: avoid human error, automatically infer generative models (#33212) 2025-02-13 16:27:11 +01:00
idefics3 CI: avoid human error, automatically infer generative models (#33212) 2025-02-13 16:27:11 +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] further fix Tester object has no attribute '_testMethodName' (#35781) 2025-01-29 16:05:33 +01:00
layoutlmv3 [tests] further fix Tester object has no attribute '_testMethodName' (#35781) 2025-01-29 16:05:33 +01:00
layoutxlm Fix #34494 assistant tokens when truncated (#34531) 2024-11-05 15:10:15 +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 use torch.testing.assertclose instead to get more details about error in cis (#35659) 2025-01-24 16:55:28 +01: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 test_export_to_onnx (#36241) 2025-02-17 16:52:44 +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 [tests] remove tf/flax tests in /generation (#36235) 2025-02-17 14:59:22 +00:00
markuplm use torch.testing.assertclose instead to get more details about error in cis (#35659) 2025-01-24 16:55:28 +01: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 Skip tests properly (#31308) 2024-06-26 21:59:08 +01: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: avoid human error, automatically infer generative models (#33212) 2025-02-13 16:27:11 +01:00
moonshine CI: avoid human error, automatically infer generative models (#33212) 2025-02-13 16:27:11 +01: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: avoid human error, automatically infer generative models (#33212) 2025-02-13 16:27:11 +01:00
musicgen_melody CI: avoid human error, automatically infer generative models (#33212) 2025-02-13 16:27:11 +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 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: avoid human error, automatically infer generative models (#33212) 2025-02-13 16:27:11 +01: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 [tests] remove test_export_to_onnx (#36241) 2025-02-17 16:52: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 [generate] remove cache v4.47 deprecations (#36212) 2025-02-17 13:55:03 +00:00
qwen2_audio [Qwen2Audio] handle input ids expansion during processing (#35534) 2025-01-07 16:47:27 +01: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: avoid human error, automatically infer generative models (#33212) 2025-02-13 16:27:11 +01: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: avoid human error, automatically infer generative models (#33212) 2025-02-13 16:27:11 +01: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: avoid human error, automatically infer generative models (#33212) 2025-02-13 16:27:11 +01:00
roformer CI: avoid human error, automatically infer generative models (#33212) 2025-02-13 16:27:11 +01: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 Handle empty change indices in SAM's mask to rle conversion (#35665) 2025-01-30 19:08:38 +00:00
seamless_m4t CI: avoid human error, automatically infer generative models (#33212) 2025-02-13 16:27:11 +01:00
seamless_m4t_v2 CI: avoid human error, automatically infer generative models (#33212) 2025-02-13 16:27:11 +01:00
segformer Add common test for torch.export and fix some vision models (#35124) 2025-02-11 11:37:31 +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
speech_encoder_decoder 🚨All attention refactor🚨 (#35235) 2024-12-18 16:53:39 +01:00
speech_to_text Add dithering to the Speech2TextFeatureExtractor API. (#34638) 2025-02-19 11:50:02 +01:00
speecht5 CI: avoid human error, automatically infer generative models (#33212) 2025-02-13 16:27:11 +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 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 test_export_to_onnx (#36241) 2025-02-17 16:52:44 +00:00
table_transformer Add common test for torch.export and fix some vision models (#35124) 2025-02-11 11:37:31 +00:00
tapas use torch.testing.assertclose instead to get more details about error in cis (#35659) 2025-01-24 16:55:28 +01: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: avoid human error, automatically infer generative models (#33212) 2025-02-13 16:27:11 +01: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 use torch.testing.assertclose instead to get more details about error in cis (#35659) 2025-01-24 16:55:28 +01:00
vision_text_dual_encoder use torch.testing.assertclose instead to get more details about error in cis (#35659) 2025-01-24 16:55:28 +01: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 Add common test for torch.export and fix some vision models (#35124) 2025-02-11 11:37:31 +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 use torch.testing.assertclose instead to get more details about error in cis (#35659) 2025-01-24 16:55:28 +01:00
wav2vec2_bert Byebye test_batching_equivalence's flakiness (#35729) 2025-01-21 13:11:33 +01:00
wav2vec2_conformer Byebye test_batching_equivalence's flakiness (#35729) 2025-01-21 13:11:33 +01: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 Add dithering to the Speech2TextFeatureExtractor API. (#34638) 2025-02-19 11:50:02 +01: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 Fix XGLM loss computation (PyTorch and TensorFlow) (#35878) 2025-02-18 15:37:48 +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 Move test model folders (#17034) 2022-05-03 14:42:02 +02:00