SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity
Files
84f0186e8971a21bcda9b446a8a74f0f1a958f1c
HuggingFace_transformer/tests/models
History
Arthur 84f0186e89
Some checks failed
Release - Conda / build_and_package (push) Has been cancelled
Details
Secret Leaks / trufflehog (push) Has been cancelled
Details
Add aya (#36521) 
* initial commit

* small fix

* move stuff to image processing file

* remove stuff in validate turn and fix return tensor

* remove liquid stuff

* in the process of addressing comments

* changes to get the right tokenization

* new __init__ works

* fixing defulat std and mean

* works

* small testing scipt -- to be deleted before merge

* remove redundant code

* addressing comments

* fix inits, add docs templates

* refactor processor, switch to gotocr image processor

* remove image proc from init

* refactor to working llava-style architecture

* Change AyaVisionModel to AyaVisionForConditionalGeneration

* add tests

* fixups

* update doc

* Adding logits_to_keep explicitly in ayavision forward to enable compatibility with cohere model

* better variable names + remove code paths

* Updates to aya_vision.md

* address comments

* adding copied from

* make style and remove unused projector_hidden_act from config

* sort init

* include usage of fast image proc and proc on cuda in doc

* update checkpoint iin test processor

* update checkpoint in test processor 2

* remove test_model and update docstring

* skip failing tests

---------

Co-authored-by: Saurabh Dash <saurabh@cohere.com>
Co-authored-by: yonigozlan <yoni.gozlan@huggingface.co>
2025-03-04 12:24:33 +01:00
..
albert
[tests] further fix Tester object has no attribute '_testMethodName' (#35781)
2025-01-29 16:05:33 +01:00
align
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
altclip
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
aria
Add @require_bitsandbytes to Aria test_batched_generation (#36192)
2025-02-14 15:48:47 +01:00
audio_spectrogram_transformer
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
auto
[tests] remove pt_tf equivalence tests (#36253)
2025-02-19 11:55:11 +00:00
autoformer
CI: avoid human error, automatically infer generative models (#33212)
2025-02-13 16:27:11 +01:00
aya_vision
Add aya (#36521)
2025-03-04 12:24:33 +01:00
bamba
CI: avoid human error, automatically infer generative models (#33212)
2025-02-13 16:27:11 +01:00
bark
CI: avoid human error, automatically infer generative models (#33212)
2025-02-13 16:27:11 +01:00
bart
[tests] remove tf/flax tests in /generation (#36235)
2025-02-17 14:59:22 +00:00
barthez
Skip tests properly (#31308)
2024-06-26 21:59:08 +01:00
bartpho
Adds pretrained IDs directly in the tests (#29534)
2024-03-13 14:53:27 +01:00
beit
Fix tests for vision models (#35654)
2025-02-13 10:28:37 +00:00
bert
CI: avoid human error, automatically infer generative models (#33212)
2025-02-13 16:27:11 +01:00
bert_generation
CI: avoid human error, automatically infer generative models (#33212)
2025-02-13 16:27:11 +01:00
bert_japanese
Adds pretrained IDs directly in the tests (#29534)
2024-03-13 14:53:27 +01:00
bertweet
Adds pretrained IDs directly in the tests (#29534)
2024-03-13 14:53:27 +01:00
big_bird
[CI] Check test if the GenerationTesterMixin inheritance is correct 🐛 🔫 (#36180)
2025-02-21 10:18:20 +00:00
bigbird_pegasus
CI: avoid human error, automatically infer generative models (#33212)
2025-02-13 16:27:11 +01:00
biogpt
CI: avoid human error, automatically infer generative models (#33212)
2025-02-13 16:27:11 +01:00
bit
Add common test for torch.export and fix some vision models (#35124)
2025-02-11 11:37:31 +00:00
blenderbot
[tests] remove tf/flax tests in /generation (#36235)
2025-02-17 14:59:22 +00:00
blenderbot_small
[tests] remove tf/flax tests in /generation (#36235)
2025-02-17 14:59:22 +00:00
blip
[CI] Check test if the GenerationTesterMixin inheritance is correct 🐛 🔫 (#36180)
2025-02-21 10:18:20 +00:00
blip_2
[CI] Check test if the GenerationTesterMixin inheritance is correct 🐛 🔫 (#36180)
2025-02-21 10:18:20 +00:00
bloom
[tests] remove tf/flax tests in /generation (#36235)
2025-02-17 14:59:22 +00:00
bridgetower
[tests] further fix Tester object has no attribute '_testMethodName' (#35781)
2025-01-29 16:05:33 +01:00
bros
CI: avoid human error, automatically infer generative models (#33212)
2025-02-13 16:27:11 +01:00
byt5
Skip tests properly (#31308)
2024-06-26 21:59:08 +01:00
camembert
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
canine
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
chameleon
CI: avoid human error, automatically infer generative models (#33212)
2025-02-13 16:27:11 +01:00
chinese_clip
[tests] further fix Tester object has no attribute '_testMethodName' (#35781)
2025-01-29 16:05:33 +01:00
clap
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
clip
[tests] remove flax-pt equivalence and cross tests (#36283)
2025-02-19 15:13:27 +00:00
clipseg
[tests] remove flax-pt equivalence and cross tests (#36283)
2025-02-19 15:13:27 +00:00
clvp
[CI] Check test if the GenerationTesterMixin inheritance is correct 🐛 🔫 (#36180)
2025-02-21 10:18:20 +00:00
code_llama
Skip tests properly (#31308)
2024-06-26 21:59:08 +01:00
codegen
CI: avoid human error, automatically infer generative models (#33212)
2025-02-13 16:27:11 +01:00
cohere
CI: avoid human error, automatically infer generative models (#33212)
2025-02-13 16:27:11 +01:00
cohere2
CI: avoid human error, automatically infer generative models (#33212)
2025-02-13 16:27:11 +01:00
colpali
Add support for nested images to LLava and VipLLava (#35558)
2025-01-30 16:49:20 -05:00
conditional_detr
[CI] Check test if the GenerationTesterMixin inheritance is correct 🐛 🔫 (#36180)
2025-02-21 10:18:20 +00:00
convbert
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
convnext
Add common test for torch.export and fix some vision models (#35124)
2025-02-11 11:37:31 +00:00
convnextv2
Add common test for torch.export and fix some vision models (#35124)
2025-02-11 11:37:31 +00:00
cpm
Make pipeline able to load processor (#32514)
2024-10-09 16:46:11 +01:00
cpmant
[CI] Check test if the GenerationTesterMixin inheritance is correct 🐛 🔫 (#36180)
2025-02-21 10:18:20 +00:00
ctrl
CI: avoid human error, automatically infer generative models (#33212)
2025-02-13 16:27:11 +01:00
cvt
Add common test for torch.export and fix some vision models (#35124)
2025-02-11 11:37:31 +00:00
dab_detr
[CI] Check test if the GenerationTesterMixin inheritance is correct 🐛 🔫 (#36180)
2025-02-21 10:18:20 +00:00
dac
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
data2vec
[tests] remove flax-pt equivalence and cross tests (#36283)
2025-02-19 15:13:27 +00:00
dbrx
CI: avoid human error, automatically infer generative models (#33212)
2025-02-13 16:27:11 +01:00
deberta
[tests] remove pt_tf equivalence tests (#36253)
2025-02-19 11:55:11 +00:00
deberta_v2
[tests] remove pt_tf equivalence tests (#36253)
2025-02-19 11:55:11 +00:00
decision_transformer
[CI] Check test if the GenerationTesterMixin inheritance is correct 🐛 🔫 (#36180)
2025-02-21 10:18:20 +00:00
deformable_detr
[CI] Check test if the GenerationTesterMixin inheritance is correct 🐛 🔫 (#36180)
2025-02-21 10:18:20 +00:00
deit
Add common test for torch.export and fix some vision models (#35124)
2025-02-11 11:37:31 +00:00
depth_anything
Add common test for torch.export and fix some vision models (#35124)
2025-02-11 11:37:31 +00:00
depth_pro
Move DataCollatorForMultipleChoice from the docs to the package (#34763)
2025-02-13 12:01:28 +01:00
detr
[CI] Check test if the GenerationTesterMixin inheritance is correct 🐛 🔫 (#36180)
2025-02-21 10:18:20 +00:00
diffllama
CI: avoid human error, automatically infer generative models (#33212)
2025-02-13 16:27:11 +01:00
dinat
Add common test for torch.export and fix some vision models (#35124)
2025-02-11 11:37:31 +00:00
dinov2
addressing the issue #34611 to make FlaxDinov2 compatible with any batch size (#35138)
2025-02-25 10:44:44 +00:00
dinov2_with_registers
Add common test for torch.export and fix some vision models (#35124)
2025-02-11 11:37:31 +00:00
distilbert
[tests] further fix Tester object has no attribute '_testMethodName' (#35781)
2025-01-29 16:05:33 +01:00
dit
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
donut
[tests] further fix Tester object has no attribute '_testMethodName' (#35781)
2025-01-29 16:05:33 +01:00
dpr
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
dpt
Add common test for torch.export and fix some vision models (#35124)
2025-02-11 11:37:31 +00:00
efficientnet
Add common test for torch.export and fix some vision models (#35124)
2025-02-11 11:37:31 +00:00
electra
[CI] Check test if the GenerationTesterMixin inheritance is correct 🐛 🔫 (#36180)
2025-02-21 10:18:20 +00:00
emu3
🔴 VLM: compile compatibility (#35724)
2025-02-14 15:23:49 +01:00
encodec
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
encoder_decoder
[tests] remove flax-pt equivalence and cross tests (#36283)
2025-02-19 15:13:27 +00:00
ernie
CI: avoid human error, automatically infer generative models (#33212)
2025-02-13 16:27:11 +01:00
esm
[tests] fix EsmModelIntegrationTest::test_inference_bitsandbytes (#36225)
2025-02-17 11:10:33 +01:00
falcon
[tests] enable bnb tests on xpu (#36233)
2025-02-24 11:30:15 +01:00
falcon_mamba
[docs] Redesign (#31757)
2025-03-03 10:33:46 -08:00
fastspeech2_conformer
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
flaubert
[CI] Check test if the GenerationTesterMixin inheritance is correct 🐛 🔫 (#36180)
2025-02-21 10:18:20 +00:00
flava
[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
Add Got-OCR 2 Fast image processor and refactor slow one (#36185)
2025-03-01 00:56:00 -05:00
gpt2
[tests] remove flax-pt equivalence and cross tests (#36283)
2025-02-19 15:13:27 +00:00
gpt_bigcode
CI: avoid human error, automatically infer generative models (#33212)
2025-02-13 16:27:11 +01:00
gpt_neo
[tests] remove flax-pt equivalence and cross tests (#36283)
2025-02-19 15:13:27 +00:00
gpt_neox
CI: avoid human error, automatically infer generative models (#33212)
2025-02-13 16:27:11 +01:00
gpt_neox_japanese
CI: avoid human error, automatically infer generative models (#33212)
2025-02-13 16:27:11 +01:00
gpt_sw3
🚨 No more default chat templates (#31733)
2024-07-24 17:36:32 +01:00
gptj
[tests] remove flax-pt equivalence and cross tests (#36283)
2025-02-19 15:13:27 +00:00
granite
CI: avoid human error, automatically infer generative models (#33212)
2025-02-13 16:27:11 +01:00
granitemoe
CI: avoid human error, automatically infer generative models (#33212)
2025-02-13 16:27:11 +01:00
granitemoeshared
add shared experts for upcoming Granite 4.0 language models (#35894)
2025-02-14 16:55:28 +01:00
grounding_dino
[GroundingDino] Fix grounding dino loss 🚨 (#31828)
2025-02-27 19:15:58 +00:00
groupvit
[tests] remove pt_tf equivalence tests (#36253)
2025-02-19 11:55:11 +00:00
helium
CI: avoid human error, automatically infer generative models (#33212)
2025-02-13 16:27:11 +01:00
herbert
Skip tests properly (#31308)
2024-06-26 21:59:08 +01:00
hiera
Add common test for torch.export and fix some vision models (#35124)
2025-02-11 11:37:31 +00:00
hubert
[tests] remove pt_tf equivalence tests (#36253)
2025-02-19 11:55:11 +00:00
ibert
[warnings] fix E721 warnings (#32223)
2024-07-25 15:12:23 +02:00
idefics
[CI] Check test if the GenerationTesterMixin inheritance is correct 🐛 🔫 (#36180)
2025-02-21 10:18:20 +00:00
idefics2
CI: avoid human error, automatically infer generative models (#33212)
2025-02-13 16:27:11 +01:00
idefics3
SmolVLM2 (#36126)
2025-02-20 15:00:26 +01:00
ijepa
Add common test for torch.export and fix some vision models (#35124)
2025-02-11 11:37:31 +00:00
imagegpt
CI: avoid human error, automatically infer generative models (#33212)
2025-02-13 16:27:11 +01:00
informer
Add missing atol to torch.testing.assert_close where rtol is specified (#36234)
2025-02-17 14:57:50 +01:00
instructblip
🔴 VLM: compile compatibility (#35724)
2025-02-14 15:23:49 +01:00
instructblipvideo
🔴 VLM: compile compatibility (#35724)
2025-02-14 15:23:49 +01:00
jamba
CI: avoid human error, automatically infer generative models (#33212)
2025-02-13 16:27:11 +01:00
jetmoe
CI: avoid human error, automatically infer generative models (#33212)
2025-02-13 16:27:11 +01:00
kosmos2
CI: avoid human error, automatically infer generative models (#33212)
2025-02-13 16:27:11 +01:00
layoutlm
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
layoutlmv2
[tests] remove pt_tf equivalence tests (#36253)
2025-02-19 11:55:11 +00:00
layoutlmv3
[tests] remove pt_tf equivalence tests (#36253)
2025-02-19 11:55:11 +00:00
layoutxlm
[tests] remove pt_tf equivalence tests (#36253)
2025-02-19 11:55:11 +00:00
led
CI: avoid human error, automatically infer generative models (#33212)
2025-02-13 16:27:11 +01:00
levit
[tests] further fix Tester object has no attribute '_testMethodName' (#35781)
2025-01-29 16:05:33 +01:00
lilt
[CI] Check test if the GenerationTesterMixin inheritance is correct 🐛 🔫 (#36180)
2025-02-21 10:18:20 +00:00
llama
[tests] remove tf/flax tests in /generation (#36235)
2025-02-17 14:59:22 +00:00
llava
Fix LlavaForConditionalGenerationModelTest::test_config after #36077 (#36230)
2025-02-17 11:49:07 +01:00
llava_next
🔴 VLM: compile compatibility (#35724)
2025-02-14 15:23:49 +01:00
llava_next_video
Uniformize LlavaNextVideoProcessor kwargs (#35613)
2025-02-18 14:13:51 -05:00
llava_onevision
🔴 VLM: compile compatibility (#35724)
2025-02-14 15:23:49 +01:00
longformer
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
longt5
[tests] remove flax-pt equivalence and cross tests (#36283)
2025-02-19 15:13:27 +00:00
luke
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
lxmert
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
m2m_100
CI: avoid human error, automatically infer generative models (#33212)
2025-02-13 16:27:11 +01:00
mamba
CI: avoid human error, automatically infer generative models (#33212)
2025-02-13 16:27:11 +01:00
mamba2
CI: avoid human error, automatically infer generative models (#33212)
2025-02-13 16:27:11 +01:00
marian
Fix broken CI on release branch due to missing conversion files (#36275)
2025-02-20 13:22:10 +01:00
markuplm
[tests] remove pt_tf equivalence tests (#36253)
2025-02-19 11:55:11 +00:00
mask2former
Add common test for torch.export and fix some vision models (#35124)
2025-02-11 11:37:31 +00:00
maskformer
Add common test for torch.export and fix some vision models (#35124)
2025-02-11 11:37:31 +00:00
mbart
[tests] remove tf/flax tests in /generation (#36235)
2025-02-17 14:59:22 +00:00
mbart50
Skip tests properly (#31308)
2024-06-26 21:59:08 +01:00
megatron_bert
[CI] Check test if the GenerationTesterMixin inheritance is correct 🐛 🔫 (#36180)
2025-02-21 10:18:20 +00:00
megatron_gpt2
Skip tests properly (#31308)
2024-06-26 21:59:08 +01:00
mgp_str
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
mimi
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
mistral
[tests] remove tf/flax tests in /generation (#36235)
2025-02-17 14:59:22 +00:00
mixtral
CI: avoid human error, automatically infer generative models (#33212)
2025-02-13 16:27:11 +01:00
mllama
CI: avoid human error, automatically infer generative models (#33212)
2025-02-13 16:27:11 +01:00
mluke
fix: Updated test_embeded_special_tokens for luke and mluke models (#32413)
2024-08-05 15:19:42 +01:00
mobilebert
MobileBERT is ExecuTorch compatible (#34473)
2024-10-29 16:14:31 +01:00
mobilenet_v1
Add common test for torch.export and fix some vision models (#35124)
2025-02-11 11:37:31 +00:00
mobilenet_v2
Add common test for torch.export and fix some vision models (#35124)
2025-02-11 11:37:31 +00:00
mobilevit
Add common test for torch.export and fix some vision models (#35124)
2025-02-11 11:37:31 +00:00
mobilevitv2
Add common test for torch.export and fix some vision models (#35124)
2025-02-11 11:37:31 +00:00
modernbert
[CI] Check test if the GenerationTesterMixin inheritance is correct 🐛 🔫 (#36180)
2025-02-21 10:18:20 +00:00
moonshine
[CI] Check test if the GenerationTesterMixin inheritance is correct 🐛 🔫 (#36180)
2025-02-21 10:18:20 +00:00
moshi
CI: avoid human error, automatically infer generative models (#33212)
2025-02-13 16:27:11 +01:00
mpnet
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
mpt
CI: avoid human error, automatically infer generative models (#33212)
2025-02-13 16:27:11 +01:00
mra
CI: avoid human error, automatically infer generative models (#33212)
2025-02-13 16:27:11 +01:00
mt5
[tests] remove test_export_to_onnx (#36241)
2025-02-17 16:52:44 +00:00
musicgen
[CI] Check test if the GenerationTesterMixin inheritance is correct 🐛 🔫 (#36180)
2025-02-21 10:18:20 +00:00
musicgen_melody
[CI] Check test if the GenerationTesterMixin inheritance is correct 🐛 🔫 (#36180)
2025-02-21 10:18:20 +00:00
mvp
CI: avoid human error, automatically infer generative models (#33212)
2025-02-13 16:27:11 +01:00
myt5
[Red CIs] Fix hub failures (#34001)
2024-10-07 10:56:24 +02:00
nemotron
CI: avoid human error, automatically infer generative models (#33212)
2025-02-13 16:27:11 +01:00
nllb
Skip tests properly (#31308)
2024-06-26 21:59:08 +01:00
nllb_moe
CI: avoid human error, automatically infer generative models (#33212)
2025-02-13 16:27:11 +01:00
nougat
[tests] further fix Tester object has no attribute '_testMethodName' (#35781)
2025-01-29 16:05:33 +01:00
nystromformer
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
olmo
CI: avoid human error, automatically infer generative models (#33212)
2025-02-13 16:27:11 +01:00
olmo2
CI: avoid human error, automatically infer generative models (#33212)
2025-02-13 16:27:11 +01:00
olmoe
CI: avoid human error, automatically infer generative models (#33212)
2025-02-13 16:27:11 +01:00
omdet_turbo
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
oneformer
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
openai
CI: avoid human error, automatically infer generative models (#33212)
2025-02-13 16:27:11 +01:00
opt
[tests] remove tf/flax tests in /generation (#36235)
2025-02-17 14:59:22 +00:00
owlv2
Uniformize OwlViT and Owlv2 processors (#35700)
2025-02-13 17:30:26 -05:00
owlvit
Uniformize OwlViT and Owlv2 processors (#35700)
2025-02-13 17:30:26 -05:00
paligemma
CI: avoid human error, automatically infer generative models (#33212)
2025-02-13 16:27:11 +01:00
paligemma2
CI: avoid human error, automatically infer generative models (#33212)
2025-02-13 16:27:11 +01:00
patchtsmixer
CI: avoid human error, automatically infer generative models (#33212)
2025-02-13 16:27:11 +01:00
patchtst
Add missing atol to torch.testing.assert_close where rtol is specified (#36234)
2025-02-17 14:57:50 +01:00
pegasus
CI: avoid human error, automatically infer generative models (#33212)
2025-02-13 16:27:11 +01:00
pegasus_x
[CI] Check test if the GenerationTesterMixin inheritance is correct 🐛 🔫 (#36180)
2025-02-21 10:18:20 +00:00
perceiver
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
persimmon
CI: avoid human error, automatically infer generative models (#33212)
2025-02-13 16:27:11 +01:00
phi
CI: avoid human error, automatically infer generative models (#33212)
2025-02-13 16:27:11 +01:00
phi3
CI: avoid human error, automatically infer generative models (#33212)
2025-02-13 16:27:11 +01:00
phimoe
[generate] remove cache v4.47 deprecations (#36212)
2025-02-17 13:55:03 +00:00
phobert
Adds pretrained IDs directly in the tests (#29534)
2024-03-13 14:53:27 +01:00
pix2struct
CI: avoid human error, automatically infer generative models (#33212)
2025-02-13 16:27:11 +01:00
pixtral
Refactoring of ImageProcessorFast (#35069)
2025-02-04 17:52:31 -05:00
plbart
CI: avoid human error, automatically infer generative models (#33212)
2025-02-13 16:27:11 +01:00
poolformer
Add common test for torch.export and fix some vision models (#35124)
2025-02-11 11:37:31 +00:00
pop2piano
[CI] Check test if the GenerationTesterMixin inheritance is correct 🐛 🔫 (#36180)
2025-02-21 10:18:20 +00:00
prophetnet
CI: avoid human error, automatically infer generative models (#33212)
2025-02-13 16:27:11 +01:00
pvt
Add common test for torch.export and fix some vision models (#35124)
2025-02-11 11:37:31 +00:00
pvt_v2
Add common test for torch.export and fix some vision models (#35124)
2025-02-11 11:37:31 +00:00
qwen2
Revert qwen2 breaking changes related to attention refactor (#36162)
2025-02-14 13:44:14 +01:00
qwen2_5_vl
[generate] remove cache v4.47 deprecations (#36212)
2025-02-17 13:55:03 +00:00
qwen2_audio
[CI] Check test if the GenerationTesterMixin inheritance is correct 🐛 🔫 (#36180)
2025-02-21 10:18:20 +00:00
qwen2_moe
CI: avoid human error, automatically infer generative models (#33212)
2025-02-13 16:27:11 +01:00
qwen2_vl
CI: avoid human error, automatically infer generative models (#33212)
2025-02-13 16:27:11 +01:00
rag
[tests] make cuda-only tests device-agnostic (#35222)
2024-12-18 10:14:22 +01:00
recurrent_gemma
[CI] Check test if the GenerationTesterMixin inheritance is correct 🐛 🔫 (#36180)
2025-02-21 10:18:20 +00:00
reformer
CI: avoid human error, automatically infer generative models (#33212)
2025-02-13 16:27:11 +01:00
regnet
Add common test for torch.export and fix some vision models (#35124)
2025-02-11 11:37:31 +00:00
rembert
[CI] Check test if the GenerationTesterMixin inheritance is correct 🐛 🔫 (#36180)
2025-02-21 10:18:20 +00:00
resnet
Add common test for torch.export and fix some vision models (#35124)
2025-02-11 11:37:31 +00:00
roberta
CI: avoid human error, automatically infer generative models (#33212)
2025-02-13 16:27:11 +01:00
roberta_prelayernorm
CI: avoid human error, automatically infer generative models (#33212)
2025-02-13 16:27:11 +01:00
roc_bert
[CI] Check test if the GenerationTesterMixin inheritance is correct 🐛 🔫 (#36180)
2025-02-21 10:18:20 +00:00
roformer
[CI] Check test if the GenerationTesterMixin inheritance is correct 🐛 🔫 (#36180)
2025-02-21 10:18:20 +00:00
rt_detr
Au revoir flaky test_fast_is_faster_than_slow (#36240)
2025-02-17 18:30:07 +01:00
rt_detr_v2
Add common test for torch.export and fix some vision models (#35124)
2025-02-11 11:37:31 +00:00
rwkv
CI: avoid human error, automatically infer generative models (#33212)
2025-02-13 16:27:11 +01:00
sam
Fix pytorch integration tests for SAM (#36397)
2025-02-25 14:53:34 +00:00
seamless_m4t
[CI] Check test if the GenerationTesterMixin inheritance is correct 🐛 🔫 (#36180)
2025-02-21 10:18:20 +00:00
seamless_m4t_v2
[CI] Check test if the GenerationTesterMixin inheritance is correct 🐛 🔫 (#36180)
2025-02-21 10:18:20 +00:00
segformer
[tests] remove pt_tf equivalence tests (#36253)
2025-02-19 11:55:11 +00:00
seggpt
Add common test for torch.export and fix some vision models (#35124)
2025-02-11 11:37:31 +00:00
sew
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
sew_d
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
siglip
Refactoring of ImageProcessorFast (#35069)
2025-02-04 17:52:31 -05:00
siglip2
Add SigLIP 2 (#36323)
2025-02-21 09:04:19 +00:00
smolvlm
SmolVLM2 (#36126)
2025-02-20 15:00:26 +01:00
speech_encoder_decoder
[tests] remove flax-pt equivalence and cross tests (#36283)
2025-02-19 15:13:27 +00:00
speech_to_text
[tests] deflake dither test (#36284)
2025-02-19 15:13:10 +00:00
speecht5
[CI] Check test if the GenerationTesterMixin inheritance is correct 🐛 🔫 (#36180)
2025-02-21 10:18:20 +00:00
splinter
Make pipeline able to load processor (#32514)
2024-10-09 16:46:11 +01:00
squeezebert
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
stablelm
CI: avoid human error, automatically infer generative models (#33212)
2025-02-13 16:27:11 +01:00
starcoder2
CI: avoid human error, automatically infer generative models (#33212)
2025-02-13 16:27:11 +01:00
superglue
CI: avoid human error, automatically infer generative models (#33212)
2025-02-13 16:27:11 +01:00
superpoint
CI: avoid human error, automatically infer generative models (#33212)
2025-02-13 16:27:11 +01:00
swiftformer
Add common test for torch.export and fix some vision models (#35124)
2025-02-11 11:37:31 +00:00
swin
Add common test for torch.export and fix some vision models (#35124)
2025-02-11 11:37:31 +00:00
swin2sr
Add common test for torch.export and fix some vision models (#35124)
2025-02-11 11:37:31 +00:00
swinv2
Add common test for torch.export and fix some vision models (#35124)
2025-02-11 11:37:31 +00:00
switch_transformers
[tests] remove test_export_to_onnx (#36241)
2025-02-17 16:52:44 +00:00
t5
[tests] remove flax-pt equivalence and cross tests (#36283)
2025-02-19 15:13:27 +00:00
table_transformer
[CI] Check test if the GenerationTesterMixin inheritance is correct 🐛 🔫 (#36180)
2025-02-21 10:18:20 +00:00
tapas
[tests] remove pt_tf equivalence tests (#36253)
2025-02-19 11:55:11 +00:00
textnet
Fix tests for vision models (#35654)
2025-02-13 10:28:37 +00:00
time_series_transformer
Add missing atol to torch.testing.assert_close where rtol is specified (#36234)
2025-02-17 14:57:50 +01:00
timesformer
Add common test for torch.export and fix some vision models (#35124)
2025-02-11 11:37:31 +00:00
timm_backbone
Less flaky for TimmBackboneModelTest::test_batching_equivalence (#35971)
2025-01-30 16:56:26 +01:00
timm_wrapper
Enable auto task for timm models in pipeline (#35531)
2025-01-08 15:14:17 +00:00
trocr
CI: avoid human error, automatically infer generative models (#33212)
2025-02-13 16:27:11 +01:00
tvp
[tests] further fix Tester object has no attribute '_testMethodName' (#35781)
2025-01-29 16:05:33 +01:00
udop
[CI] Check test if the GenerationTesterMixin inheritance is correct 🐛 🔫 (#36180)
2025-02-21 10:18:20 +00:00
umt5
[tests] remove test_export_to_onnx (#36241)
2025-02-17 16:52:44 +00:00
unispeech
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
unispeech_sat
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
univnet
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
upernet
Add common test for torch.export and fix some vision models (#35124)
2025-02-11 11:37:31 +00:00
video_llava
🔴 VLM: compile compatibility (#35724)
2025-02-14 15:23:49 +01:00
videomae
Add common test for torch.export and fix some vision models (#35124)
2025-02-11 11:37:31 +00:00
vilt
[tests] further fix Tester object has no attribute '_testMethodName' (#35781)
2025-01-29 16:05:33 +01:00
vipllava
🔴 VLM: compile compatibility (#35724)
2025-02-14 15:23:49 +01:00
vision_encoder_decoder
[tests] remove flax-pt equivalence and cross tests (#36283)
2025-02-19 15:13:27 +00:00
vision_text_dual_encoder
[tests] remove flax-pt equivalence and cross tests (#36283)
2025-02-19 15:13:27 +00:00
visual_bert
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
vit
Fix tests for vision models (#35654)
2025-02-13 10:28:37 +00:00
vit_mae
[tests] remove pt_tf equivalence tests (#36253)
2025-02-19 11:55:11 +00:00
vit_msn
Add common test for torch.export and fix some vision models (#35124)
2025-02-11 11:37:31 +00:00
vitdet
Fixed VitDet for non-squre Images (#35969)
2025-02-25 19:31:24 +00:00
vitmatte
Add common test for torch.export and fix some vision models (#35124)
2025-02-11 11:37:31 +00:00
vitpose
Add common test for torch.export and fix some vision models (#35124)
2025-02-11 11:37:31 +00:00
vitpose_backbone
skip test_initialization for VitPoseBackboneModelTest for now (#36154)
2025-02-12 18:24:24 +01:00
vits
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
vivit
Add common test for torch.export and fix some vision models (#35124)
2025-02-11 11:37:31 +00:00
wav2vec2
[tests] remove flax-pt equivalence and cross tests (#36283)
2025-02-19 15:13:27 +00:00
wav2vec2_bert
[tests] remove flax-pt equivalence and cross tests (#36283)
2025-02-19 15:13:27 +00:00
wav2vec2_conformer
[tests] remove flax-pt equivalence and cross tests (#36283)
2025-02-19 15:13:27 +00:00
wav2vec2_phoneme
clean_up_tokenization_spaces=False if unset (#31938)
2024-09-26 19:38:20 +02:00
wav2vec2_with_lm
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
wavlm
Add missing atol to torch.testing.assert_close where rtol is specified (#36234)
2025-02-17 14:57:50 +01:00
whisper
[CI] Check test if the GenerationTesterMixin inheritance is correct 🐛 🔫 (#36180)
2025-02-21 10:18:20 +00:00
x_clip
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
xglm
[tests] remove flax-pt equivalence and cross tests (#36283)
2025-02-19 15:13:27 +00:00
xlm
CI: avoid human error, automatically infer generative models (#33212)
2025-02-13 16:27:11 +01:00
xlm_roberta
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
xlm_roberta_xl
CI: avoid human error, automatically infer generative models (#33212)
2025-02-13 16:27:11 +01:00
xlnet
CI: avoid human error, automatically infer generative models (#33212)
2025-02-13 16:27:11 +01:00
xmod
CI: avoid human error, automatically infer generative models (#33212)
2025-02-13 16:27:11 +01:00
yolos
Add common test for torch.export and fix some vision models (#35124)
2025-02-11 11:37:31 +00:00
yoso
CI: avoid human error, automatically infer generative models (#33212)
2025-02-13 16:27:11 +01:00
zamba
CI: avoid human error, automatically infer generative models (#33212)
2025-02-13 16:27:11 +01:00
zamba2
CI: avoid human error, automatically infer generative models (#33212)
2025-02-13 16:27:11 +01:00
zoedepth
Fix tests for vision models (#35654)
2025-02-13 10:28:37 +00:00
__init__.py
Move test model folders (#17034)
2022-05-03 14:42:02 +02:00