transformers/tests/models
Joao Gante be2ac0916a
[generate] shape checks in tests compatible with fixed-length caches (+ some minor fixes) (#35993)
* shape checks compatible with static cache

* add test

* tmp

* manually turn on eager attn when we want to output attn

* typo

* generalize to encoder-decoder models

* force compilation on cpu

* tmp commit

* fix static cache shape checks

* models with odd caches

* fix copies

* shorter cache search loop

* use decoder_past_key_values everywhere

* better test variable names and comments

* signature

* rename _check_outputs into _check_generate_outputs

* add comments

* HybridCache future test note
2025-02-10 17:50:54 +00: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 Chat template: update for processor (#35953) 2025-02-10 09:52:19 +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 use torch.testing.assertclose instead to get more details about error in cis (#35659) 2025-01-24 16:55:28 +01:00
bamba use torch.testing.assertclose instead to get more details about error in cis (#35659) 2025-01-24 16:55:28 +01:00
bark use torch.testing.assertclose instead to get more details about error in cis (#35659) 2025-01-24 16:55:28 +01:00
bart [bart] minor test fixes (#35965) 2025-01-30 10:00:11 +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 [tests] further fix Tester object has no attribute '_testMethodName' (#35781) 2025-01-29 16:05:33 +01:00
bert [tests] further fix Tester object has no attribute '_testMethodName' (#35781) 2025-01-29 16:05:33 +01:00
bert_generation use torch.testing.assertclose instead to get more details about error in cis (#35659) 2025-01-24 16:55:28 +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 [tests] further fix Tester object has no attribute '_testMethodName' (#35781) 2025-01-29 16:05:33 +01:00
bigbird_pegasus use torch.testing.assertclose instead to get more details about error in cis (#35659) 2025-01-24 16:55:28 +01:00
biogpt use torch.testing.assertclose instead to get more details about error in cis (#35659) 2025-01-24 16:55:28 +01:00
bit use torch.testing.assertclose instead to get more details about error in cis (#35659) 2025-01-24 16:55:28 +01:00
blenderbot Forbid PretrainedConfig from saving generate parameters; Update deprecations in generate-related code 🧹 (#32659) 2024-08-23 11:12:53 +01:00
blenderbot_small Make pipeline able to load processor (#32514) 2024-10-09 16:46:11 +01:00
blip Refactoring of ImageProcessorFast (#35069) 2025-02-04 17:52:31 -05:00
blip_2 [generate] shape checks in tests compatible with fixed-length caches (+ some minor fixes) (#35993) 2025-02-10 17:50:54 +00:00
bloom VLMs: enable generation tests (#33533) 2024-09-19 12:04:24 +02:00
bridgetower [tests] further fix Tester object has no attribute '_testMethodName' (#35781) 2025-01-29 16:05:33 +01:00
bros use torch.testing.assertclose instead to get more details about error in cis (#35659) 2025-01-24 16:55:28 +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 [tests] further fix Tester object has no attribute '_testMethodName' (#35781) 2025-01-29 16:05:33 +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 Iterative generation using Input embeds and past_key_values (#35890) 2025-02-06 11:06:05 +01:00
code_llama Skip tests properly (#31308) 2024-06-26 21:59:08 +01:00
codegen [generate] move max time tests (#35962) 2025-01-29 17:56:46 +00:00
cohere use torch.testing.assertclose instead to get more details about error in cis (#35659) 2025-01-24 16:55:28 +01:00
cohere2 [generate] shape checks in tests compatible with fixed-length caches (+ some minor fixes) (#35993) 2025-02-10 17:50:54 +00:00
colpali Add support for nested images to LLava and VipLLava (#35558) 2025-01-30 16:49:20 -05:00
conditional_detr use torch.testing.assertclose instead to get more details about error in cis (#35659) 2025-01-24 16:55:28 +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 Refactoring of ImageProcessorFast (#35069) 2025-02-04 17:52:31 -05:00
convnextv2 use torch.testing.assertclose instead to get more details about error in cis (#35659) 2025-01-24 16:55:28 +01: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 avoid calling gc.collect and cuda.empty_cache (#34514) 2024-10-31 16:36:13 +01:00
cvt use torch.testing.assertclose instead to get more details about error in cis (#35659) 2025-01-24 16:55:28 +01:00
dab_detr Add DAB-DETR for object detection (#30803) 2025-02-04 17:28:27 +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 use torch.testing.assertclose instead to get more details about error in cis (#35659) 2025-01-24 16:55:28 +01:00
dbrx Test: generate with torch.compile(model.forward) as a fast test (#34544) 2025-01-28 14:10:38 +00: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 use torch.testing.assertclose instead to get more details about error in cis (#35659) 2025-01-24 16:55:28 +01:00
deformable_detr [tests] further fix Tester object has no attribute '_testMethodName' (#35781) 2025-01-29 16:05:33 +01:00
deit Refactoring of ImageProcessorFast (#35069) 2025-02-04 17:52:31 -05:00
depth_anything use torch.testing.assertclose instead to get more details about error in cis (#35659) 2025-01-24 16:55:28 +01:00
depth_pro Add Apple's Depth-Pro for depth estimation (#34583) 2025-02-10 11:32:45 +00:00
detr [tests] further fix Tester object has no attribute '_testMethodName' (#35781) 2025-01-29 16:05:33 +01:00
diffllama use torch.testing.assertclose instead to get more details about error in cis (#35659) 2025-01-24 16:55:28 +01:00
dinat use torch.testing.assertclose instead to get more details about error in cis (#35659) 2025-01-24 16:55:28 +01:00
dinov2 use torch.testing.assertclose instead to get more details about error in cis (#35659) 2025-01-24 16:55:28 +01:00
dinov2_with_registers use torch.testing.assertclose instead to get more details about error in cis (#35659) 2025-01-24 16:55:28 +01: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 [tests] further fix Tester object has no attribute '_testMethodName' (#35781) 2025-01-29 16:05:33 +01:00
efficientnet [tests] further fix Tester object has no attribute '_testMethodName' (#35781) 2025-01-29 16:05:33 +01:00
electra [tests] further fix Tester object has no attribute '_testMethodName' (#35781) 2025-01-29 16:05:33 +01:00
emu3 Chat template: update for processor (#35953) 2025-02-10 09:52:19 +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 Skip tests properly (#31308) 2024-06-26 21:59:08 +01:00
esm use torch.testing.assertclose instead to get more details about error in cis (#35659) 2025-01-24 16:55:28 +01:00
falcon use torch.testing.assertclose instead to get more details about error in cis (#35659) 2025-01-24 16:55:28 +01:00
falcon_mamba use torch.testing.assertclose instead to get more details about error in cis (#35659) 2025-01-24 16:55:28 +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 use torch.testing.assertclose instead to get more details about error in cis (#35659) 2025-01-24 16:55:28 +01:00
fsmt use torch.testing.assertclose instead to get more details about error in cis (#35659) 2025-01-24 16:55:28 +01: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 Iterative generation using Input embeds and past_key_values (#35890) 2025-02-06 11:06:05 +01:00
gemma Tests: move generate tests to the right mixin and delete redundant tests (#34464) 2024-10-30 10:59:08 +00:00
gemma2 [generate] shape checks in tests compatible with fixed-length caches (+ some minor fixes) (#35993) 2025-02-10 17:50:54 +00:00
git [generate] shape checks in tests compatible with fixed-length caches (+ some minor fixes) (#35993) 2025-02-10 17:50:54 +00:00
glm Fix some tests (#35682) 2025-01-17 12:10:43 +00:00
glpn [tests] further fix Tester object has no attribute '_testMethodName' (#35781) 2025-01-29 16:05:33 +01:00
got_ocr2 Add GOT-OCR 2.0 to Transformers (#34721) 2025-01-31 11:28:13 -05:00
gpt_bigcode Iterative generation using Input embeds and past_key_values (#35890) 2025-02-06 11:06:05 +01:00
gpt_neo Generation tests: update imagegpt input name, remove unused functions (#33663) 2024-09-24 16:40:48 +01:00
gpt_neox Refactor (and fix) gpt_neox (#35610) 2025-02-04 11:18:43 +01:00
gpt_neox_japanese Compile compatibilty for decoder-only models (#32617) 2024-09-09 10:59:04 +02:00
gpt_sw3 🚨 No more default chat templates (#31733) 2024-07-24 17:36:32 +01:00
gpt2 [generate] move max time tests (#35962) 2025-01-29 17:56:46 +00:00
gptj [generate] move max time tests (#35962) 2025-01-29 17:56:46 +00:00
granite use torch.testing.assertclose instead to get more details about error in cis (#35659) 2025-01-24 16:55:28 +01:00
granitemoe use torch.testing.assertclose instead to get more details about error in cis (#35659) 2025-01-24 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 Add-helium (#35669) 2025-01-13 18:41:15 +01:00
herbert Skip tests properly (#31308) 2024-06-26 21:59:08 +01:00
hiera use torch.testing.assertclose instead to get more details about error in cis (#35659) 2025-01-24 16:55:28 +01: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 [generate] shape checks in tests compatible with fixed-length caches (+ some minor fixes) (#35993) 2025-02-10 17:50:54 +00:00
idefics2 use torch.testing.assertclose instead to get more details about error in cis (#35659) 2025-01-24 16:55:28 +01:00
idefics3 [tests] further fix Tester object has no attribute '_testMethodName' (#35781) 2025-01-29 16:05:33 +01:00
ijepa use torch.testing.assertclose instead to get more details about error in cis (#35659) 2025-01-24 16:55:28 +01:00
imagegpt [generate] shape checks in tests compatible with fixed-length caches (+ some minor fixes) (#35993) 2025-02-10 17:50:54 +00:00
informer use torch.testing.assertclose instead to get more details about error in cis (#35659) 2025-01-24 16:55:28 +01:00
instructblip [generate] shape checks in tests compatible with fixed-length caches (+ some minor fixes) (#35993) 2025-02-10 17:50:54 +00:00
instructblipvideo [generate] shape checks in tests compatible with fixed-length caches (+ some minor fixes) (#35993) 2025-02-10 17:50:54 +00:00
jamba use torch.testing.assertclose instead to get more details about error in cis (#35659) 2025-01-24 16:55:28 +01:00
jetmoe use torch.testing.assertclose instead to get more details about error in cis (#35659) 2025-01-24 16:55:28 +01:00
kosmos2 use torch.testing.assertclose instead to get more details about error in cis (#35659) 2025-01-24 16:55:28 +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 [generate] shape checks in tests compatible with fixed-length caches (+ some minor fixes) (#35993) 2025-02-10 17:50:54 +00: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 use torch.testing.assertclose instead to get more details about error in cis (#35659) 2025-01-24 16:55:28 +01:00
llava Chat template: update for processor (#35953) 2025-02-10 09:52:19 +01:00
llava_next Chat template: update for processor (#35953) 2025-02-10 09:52:19 +01:00
llava_next_video Refactoring of ImageProcessorFast (#35069) 2025-02-04 17:52:31 -05:00
llava_onevision Chat template: update for processor (#35953) 2025-02-10 09:52:19 +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 [generate] shape checks in tests compatible with fixed-length caches (+ some minor fixes) (#35993) 2025-02-10 17:50:54 +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 use torch.testing.assertclose instead to get more details about error in cis (#35659) 2025-01-24 16:55:28 +01:00
mamba use torch.testing.assertclose instead to get more details about error in cis (#35659) 2025-01-24 16:55:28 +01:00
mamba2 use torch.testing.assertclose instead to get more details about error in cis (#35659) 2025-01-24 16:55:28 +01:00
marian Forbid PretrainedConfig from saving generate parameters; Update deprecations in generate-related code 🧹 (#32659) 2024-08-23 11:12:53 +01: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 use torch.testing.assertclose instead to get more details about error in cis (#35659) 2025-01-24 16:55:28 +01:00
maskformer use torch.testing.assertclose instead to get more details about error in cis (#35659) 2025-01-24 16:55:28 +01:00
mbart [tests] further fix Tester object has no attribute '_testMethodName' (#35781) 2025-01-29 16:05:33 +01: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 use torch.testing.assertclose instead to get more details about error in cis (#35659) 2025-01-24 16:55:28 +01:00
mixtral [tests] make cuda-only tests device-agnostic (#35607) 2025-01-13 14:48:39 +01:00
mllama [generate] shape checks in tests compatible with fixed-length caches (+ some minor fixes) (#35993) 2025-02-10 17:50:54 +00: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 [tests] further fix Tester object has no attribute '_testMethodName' (#35781) 2025-01-29 16:05:33 +01:00
mobilenet_v2 [tests] further fix Tester object has no attribute '_testMethodName' (#35781) 2025-01-29 16:05:33 +01:00
mobilevit [tests] further fix Tester object has no attribute '_testMethodName' (#35781) 2025-01-29 16:05:33 +01:00
mobilevitv2 use torch.testing.assertclose instead to get more details about error in cis (#35659) 2025-01-24 16:55:28 +01:00
modernbert use torch.testing.assertclose instead to get more details about error in cis (#35659) 2025-01-24 16:55:28 +01:00
moonshine use torch.testing.assertclose instead to get more details about error in cis (#35659) 2025-01-24 16:55:28 +01:00
moshi [generate] shape checks in tests compatible with fixed-length caches (+ some minor fixes) (#35993) 2025-02-10 17:50:54 +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 use torch.testing.assertclose instead to get more details about error in cis (#35659) 2025-01-24 16:55:28 +01:00
mra use torch.testing.assertclose instead to get more details about error in cis (#35659) 2025-01-24 16:55:28 +01:00
mt5 Tests: move generate tests to the right mixin and delete redundant tests (#34464) 2024-10-30 10:59:08 +00:00
musicgen use torch.testing.assertclose instead to get more details about error in cis (#35659) 2025-01-24 16:55:28 +01:00
musicgen_melody use torch.testing.assertclose instead to get more details about error in cis (#35659) 2025-01-24 16:55:28 +01:00
mvp use torch.testing.assertclose instead to get more details about error in cis (#35659) 2025-01-24 16:55:28 +01:00
myt5 [Red CIs] Fix hub failures (#34001) 2024-10-07 10:56:24 +02:00
nemotron [tests] make cuda-only tests device-agnostic (#35607) 2025-01-13 14:48:39 +01:00
nllb Skip tests properly (#31308) 2024-06-26 21:59:08 +01:00
nllb_moe use torch.testing.assertclose instead to get more details about error in cis (#35659) 2025-01-24 16:55:28 +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 use torch.testing.assertclose instead to get more details about error in cis (#35659) 2025-01-24 16:55:28 +01:00
olmo2 use torch.testing.assertclose instead to get more details about error in cis (#35659) 2025-01-24 16:55:28 +01:00
olmoe use torch.testing.assertclose instead to get more details about error in cis (#35659) 2025-01-24 16:55:28 +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 Make pipeline able to load processor (#32514) 2024-10-09 16:46:11 +01:00
opt Tests: upgrade test_eager_matches_sdpa_generate (#34386) 2024-10-25 11:55:07 +01:00
owlv2 [tests] further fix Tester object has no attribute '_testMethodName' (#35781) 2025-01-29 16:05:33 +01:00
owlvit [tests] further fix Tester object has no attribute '_testMethodName' (#35781) 2025-01-29 16:05:33 +01:00
paligemma use torch.testing.assertclose instead to get more details about error in cis (#35659) 2025-01-24 16:55:28 +01:00
paligemma2 Paligemma: fix generation with Gemma2 (#36044) 2025-02-06 14:31:32 +01:00
patchtsmixer use torch.testing.assertclose instead to get more details about error in cis (#35659) 2025-01-24 16:55:28 +01:00
patchtst use torch.testing.assertclose instead to get more details about error in cis (#35659) 2025-01-24 16:55:28 +01:00
pegasus Forbid PretrainedConfig from saving generate parameters; Update deprecations in generate-related code 🧹 (#32659) 2024-08-23 11:12:53 +01:00
pegasus_x [generate] shape checks in tests compatible with fixed-length caches (+ some minor fixes) (#35993) 2025-02-10 17:50:54 +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 use torch.testing.assertclose instead to get more details about error in cis (#35659) 2025-01-24 16:55:28 +01:00
phi use torch.testing.assertclose instead to get more details about error in cis (#35659) 2025-01-24 16:55:28 +01:00
phi3 use torch.testing.assertclose instead to get more details about error in cis (#35659) 2025-01-24 16:55:28 +01:00
phimoe use torch.testing.assertclose instead to get more details about error in cis (#35659) 2025-01-24 16:55:28 +01:00
phobert Adds pretrained IDs directly in the tests (#29534) 2024-03-13 14:53:27 +01:00
pix2struct [generate] shape checks in tests compatible with fixed-length caches (+ some minor fixes) (#35993) 2025-02-10 17:50:54 +00:00
pixtral Refactoring of ImageProcessorFast (#35069) 2025-02-04 17:52:31 -05:00
plbart Make pipeline able to load processor (#32514) 2024-10-09 16:46:11 +01:00
poolformer [tests] further fix Tester object has no attribute '_testMethodName' (#35781) 2025-01-29 16:05:33 +01:00
pop2piano use torch.testing.assertclose instead to get more details about error in cis (#35659) 2025-01-24 16:55:28 +01:00
prophetnet use torch.testing.assertclose instead to get more details about error in cis (#35659) 2025-01-24 16:55:28 +01:00
pvt [tests] further fix Tester object has no attribute '_testMethodName' (#35781) 2025-01-29 16:05:33 +01:00
pvt_v2 use torch.testing.assertclose instead to get more details about error in cis (#35659) 2025-01-24 16:55:28 +01:00
qwen2 use torch.testing.assertclose instead to get more details about error in cis (#35659) 2025-01-24 16:55:28 +01:00
qwen2_5_vl Chat template: update for processor (#35953) 2025-02-10 09:52:19 +01:00
qwen2_audio [Qwen2Audio] handle input ids expansion during processing (#35534) 2025-01-07 16:47:27 +01:00
qwen2_moe use torch.testing.assertclose instead to get more details about error in cis (#35659) 2025-01-24 16:55:28 +01:00
qwen2_vl Chat template: update for processor (#35953) 2025-02-10 09:52:19 +01:00
rag [tests] make cuda-only tests device-agnostic (#35222) 2024-12-18 10:14:22 +01:00
recurrent_gemma [generate] shape checks in tests compatible with fixed-length caches (+ some minor fixes) (#35993) 2025-02-10 17:50:54 +00:00
reformer [generate] shape checks in tests compatible with fixed-length caches (+ some minor fixes) (#35993) 2025-02-10 17:50:54 +00:00
regnet [tests] further fix Tester object has no attribute '_testMethodName' (#35781) 2025-01-29 16:05:33 +01:00
rembert use torch.testing.assertclose instead to get more details about error in cis (#35659) 2025-01-24 16:55:28 +01:00
resnet [tests] further fix Tester object has no attribute '_testMethodName' (#35781) 2025-01-29 16:05:33 +01:00
roberta [tests] further fix Tester object has no attribute '_testMethodName' (#35781) 2025-01-29 16:05:33 +01:00
roberta_prelayernorm [tests] further fix Tester object has no attribute '_testMethodName' (#35781) 2025-01-29 16:05:33 +01:00
roc_bert Make pipeline able to load processor (#32514) 2024-10-09 16:46:11 +01:00
roformer [tests] further fix Tester object has no attribute '_testMethodName' (#35781) 2025-01-29 16:05:33 +01:00
rt_detr [tests] further fix Tester object has no attribute '_testMethodName' (#35781) 2025-01-29 16:05:33 +01:00
rt_detr_v2 Adding RT-DETRv2 for object detection (#34773) 2025-02-06 19:28:45 +00:00
rwkv use torch.testing.assertclose instead to get more details about error in cis (#35659) 2025-01-24 16:55:28 +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 use torch.testing.assertclose instead to get more details about error in cis (#35659) 2025-01-24 16:55:28 +01:00
seamless_m4t_v2 Make test_generate_with_static_cache even less flaky (#34995) 2024-12-20 16:03:26 +01:00
segformer use torch.testing.assertclose instead to get more details about error in cis (#35659) 2025-01-24 16:55:28 +01:00
seggpt use torch.testing.assertclose instead to get more details about error in cis (#35659) 2025-01-24 16:55:28 +01: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 [generate] shape checks in tests compatible with fixed-length caches (+ some minor fixes) (#35993) 2025-02-10 17:50:54 +00:00
speecht5 use torch.testing.assertclose instead to get more details about error in cis (#35659) 2025-01-24 16:55:28 +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 use torch.testing.assertclose instead to get more details about error in cis (#35659) 2025-01-24 16:55:28 +01:00
starcoder2 [tests] make cuda-only tests device-agnostic (#35607) 2025-01-13 14:48:39 +01:00
superglue [tests] further fix Tester object has no attribute '_testMethodName' (#35781) 2025-01-29 16:05:33 +01:00
superpoint use torch.testing.assertclose instead to get more details about error in cis (#35659) 2025-01-24 16:55:28 +01:00
swiftformer use torch.testing.assertclose instead to get more details about error in cis (#35659) 2025-01-24 16:55:28 +01:00
swin use torch.testing.assertclose instead to get more details about error in cis (#35659) 2025-01-24 16:55:28 +01:00
swin2sr [tests] further fix Tester object has no attribute '_testMethodName' (#35781) 2025-01-29 16:05:33 +01:00
swinv2 use torch.testing.assertclose instead to get more details about error in cis (#35659) 2025-01-24 16:55:28 +01:00
switch_transformers Fix flaky SwitchTransformersModelTest::test_training_gradient (#35587) 2025-01-09 15:36:22 +01:00
t5 use torch.testing.assertclose instead to get more details about error in cis (#35659) 2025-01-24 16:55:28 +01:00
table_transformer use torch.testing.assertclose instead to get more details about error in cis (#35659) 2025-01-24 16:55:28 +01: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 [tests] further fix Tester object has no attribute '_testMethodName' (#35781) 2025-01-29 16:05:33 +01:00
time_series_transformer use torch.testing.assertclose instead to get more details about error in cis (#35659) 2025-01-24 16:55:28 +01:00
timesformer use torch.testing.assertclose instead to get more details about error in cis (#35659) 2025-01-24 16:55:28 +01: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 🚨🚨🚨 Uniformize kwargs for TrOCR Processor (#34587) 2024-11-29 11:58:11 +00:00
tvp [tests] further fix Tester object has no attribute '_testMethodName' (#35781) 2025-01-29 16:05:33 +01:00
udop Fix #34494 assistant tokens when truncated (#34531) 2024-11-05 15:10:15 +00:00
umt5 Tests: move generate tests to the right mixin and delete redundant tests (#34464) 2024-10-30 10:59:08 +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 use torch.testing.assertclose instead to get more details about error in cis (#35659) 2025-01-24 16:55:28 +01:00
video_llava Refactoring of ImageProcessorFast (#35069) 2025-02-04 17:52:31 -05:00
videomae [tests] further fix Tester object has no attribute '_testMethodName' (#35781) 2025-01-29 16:05:33 +01:00
vilt [tests] further fix Tester object has no attribute '_testMethodName' (#35781) 2025-01-29 16:05:33 +01:00
vipllava use torch.testing.assertclose instead to get more details about error in cis (#35659) 2025-01-24 16:55:28 +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 Refactoring of ImageProcessorFast (#35069) 2025-02-04 17:52:31 -05:00
vit_mae use torch.testing.assertclose instead to get more details about error in cis (#35659) 2025-01-24 16:55:28 +01:00
vit_msn use torch.testing.assertclose instead to get more details about error in cis (#35659) 2025-01-24 16:55:28 +01:00
vitdet Skip tests properly (#31308) 2024-06-26 21:59:08 +01:00
vitmatte [tests] further fix Tester object has no attribute '_testMethodName' (#35781) 2025-01-29 16:05:33 +01:00
vitpose [tests] further fix Tester object has no attribute '_testMethodName' (#35781) 2025-01-29 16:05:33 +01:00
vitpose_backbone Add ViTPose (#30530) 2025-01-08 16:02:14 +00: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 [tests] further fix Tester object has no attribute '_testMethodName' (#35781) 2025-01-29 16:05:33 +01: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 use torch.testing.assertclose instead to get more details about error in cis (#35659) 2025-01-24 16:55:28 +01:00
whisper [generate] shape checks in tests compatible with fixed-length caches (+ some minor fixes) (#35993) 2025-02-10 17:50:54 +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 [generate] move max time tests (#35962) 2025-01-29 17:56:46 +00:00
xlm [generate] shape checks in tests compatible with fixed-length caches (+ some minor fixes) (#35993) 2025-02-10 17:50:54 +00: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 use torch.testing.assertclose instead to get more details about error in cis (#35659) 2025-01-24 16:55:28 +01:00
xlnet [generate] shape checks in tests compatible with fixed-length caches (+ some minor fixes) (#35993) 2025-02-10 17:50:54 +00:00
xmod use torch.testing.assertclose instead to get more details about error in cis (#35659) 2025-01-24 16:55:28 +01:00
yolos use torch.testing.assertclose instead to get more details about error in cis (#35659) 2025-01-24 16:55:28 +01:00
yoso use torch.testing.assertclose instead to get more details about error in cis (#35659) 2025-01-24 16:55:28 +01:00
zamba use torch.testing.assertclose instead to get more details about error in cis (#35659) 2025-01-24 16:55:28 +01:00
zamba2 Iterative generation using Input embeds and past_key_values (#35890) 2025-02-06 11:06:05 +01:00
zoedepth [tests] further fix Tester object has no attribute '_testMethodName' (#35781) 2025-01-29 16:05:33 +01:00
__init__.py Move test model folders (#17034) 2022-05-03 14:42:02 +02:00