SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity
Files
6acd5aecb34b579729c29a26aa470f92081d88fa
HuggingFace_transformer/tests/models
History
Bo Zheng 6acd5aecb3 Adding Qwen3 and Qwen3MoE (#36878) 
* Initial commit for Qwen3

* fix and add tests for qwen3 & qwen3_moe

* rename models for tests.

* fix

* fix

* fix and add docs.

* fix model name in docs.

* simplify modular and fix configuration issues

* Fix the red CI: ruff was updated

* revert ruff, version was wrong

* fix qwen3moe.

* fix

* make sure MOE can load

* fix copies

---------

Co-authored-by: Arthur Zucker <arthur.zucker@gmail.com>
2025-03-31 09:50:49 +02:00
..
albert
Use lru_cache for tokenization tests (#36818)
2025-03-28 15:09:35 +01:00
align
chore: fix typos in tests directory (#36785)
2025-03-18 10:31:13 +01:00
altclip
chore: fix typos in tests directory (#36785)
2025-03-18 10:31:13 +01:00
aria
Support return_tensors in audio chat templates (#34601)
2025-03-25 11:08:47 +01:00
audio_spectrogram_transformer
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
auto
chore: fix typos in tests directory (#36785)
2025-03-18 10:31:13 +01:00
autoformer
Set weights_only in torch.load (#36991)
2025-03-27 14:55:50 +00:00
aya_vision
[Generation, Gemma 3] When passing a custom generation_config, overwrite default values with the model's base generation_config (#36684)
2025-03-15 12:40:09 +00:00
bamba
Expectations test utils (#36569)
2025-03-18 23:39:50 +01:00
bark
fix can_generate (#36570)
2025-03-17 14:56:18 +01:00
bart
[generate] beam search -- fix output cropping (#37080)
2025-03-28 18:57:51 +01:00
barthez
Use lru_cache for tokenization tests (#36818)
2025-03-28 15:09:35 +01:00
bartpho
Use lru_cache for tokenization tests (#36818)
2025-03-28 15:09:35 +01:00
beit
[tests] Parameterized test_eager_matches_sdpa_inference (#36650)
2025-03-14 14:41:27 +00:00
bert
Use lru_cache for tokenization tests (#36818)
2025-03-28 15:09:35 +01:00
bert_generation
Use lru_cache for tokenization tests (#36818)
2025-03-28 15:09:35 +01:00
bert_japanese
Use lru_cache for tokenization tests (#36818)
2025-03-28 15:09:35 +01:00
bertweet
Use lru_cache for tokenization tests (#36818)
2025-03-28 15:09:35 +01:00
big_bird
Use lru_cache for tokenization tests (#36818)
2025-03-28 15:09:35 +01:00
bigbird_pegasus
Fix typos in tests (#36547)
2025-03-05 15:04:06 -08:00
biogpt
[generate] beam search -- fix output cropping (#37080)
2025-03-28 18:57:51 +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
Use lru_cache for tokenization tests (#36818)
2025-03-28 15:09:35 +01:00
blip
chore: fix typos in the tests directory (#36813)
2025-03-21 10:20:05 +01:00
blip_2
chore: fix typos in the tests directory (#36813)
2025-03-21 10:20:05 +01:00
bloom
Use lru_cache for tokenization tests (#36818)
2025-03-28 15:09:35 +01:00
bridgetower
[tests] further fix Tester object has no attribute '_testMethodName' (#35781)
2025-01-29 16:05:33 +01:00
bros
CI: avoid human error, automatically infer generative models (#33212)
2025-02-13 16:27:11 +01:00
byt5
Use lru_cache for tokenization tests (#36818)
2025-03-28 15:09:35 +01:00
camembert
Use lru_cache for tokenization tests (#36818)
2025-03-28 15:09:35 +01:00
canine
Use lru_cache for tokenization tests (#36818)
2025-03-28 15:09:35 +01:00
chameleon
fix typos in the tests directory (#36932)
2025-03-25 10:49:24 +01:00
chinese_clip
chore: fix typos in the tests directory (#36813)
2025-03-21 10:20:05 +01:00
clap
chore: fix typos in the tests directory (#36813)
2025-03-21 10:20:05 +01:00
clip
Use lru_cache for tokenization tests (#36818)
2025-03-28 15:09:35 +01:00
clipseg
chore: fix typos in the tests directory (#36813)
2025-03-21 10:20:05 +01:00
clvp
Use lru_cache for tokenization tests (#36818)
2025-03-28 15:09:35 +01:00
code_llama
Use lru_cache for tokenization tests (#36818)
2025-03-28 15:09:35 +01:00
codegen
Use lru_cache for tokenization tests (#36818)
2025-03-28 15:09:35 +01:00
cohere
Use lru_cache for tokenization tests (#36818)
2025-03-28 15:09:35 +01:00
cohere2
[generate] ✨ vectorized beam search ✨ (#35802)
2025-03-18 18:39:36 +00:00
colpali
[tests] Parameterized test_eager_matches_sdpa_inference (#36650)
2025-03-14 14:41:27 +00:00
conditional_detr
chore: fix typos in the tests directory (#36813)
2025-03-21 10:20:05 +01:00
convbert
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
convnext
Add common test for torch.export and fix some vision models (#35124)
2025-02-11 11:37:31 +00:00
convnextv2
Add common test for torch.export and fix some vision models (#35124)
2025-02-11 11:37:31 +00:00
cpm
Make pipeline able to load processor (#32514)
2024-10-09 16:46:11 +01:00
cpmant
Use lru_cache for tokenization tests (#36818)
2025-03-28 15:09:35 +01:00
ctrl
Use lru_cache for tokenization tests (#36818)
2025-03-28 15:09:35 +01:00
cvt
Add common test for torch.export and fix some vision models (#35124)
2025-02-11 11:37:31 +00:00
dab_detr
fix typos in the tests directory (#36932)
2025-03-25 10:49:24 +01:00
dac
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
data2vec
[tests] Parameterized test_eager_matches_sdpa_inference (#36650)
2025-03-14 14:41:27 +00:00
dbrx
CI: avoid human error, automatically infer generative models (#33212)
2025-02-13 16:27:11 +01:00
deberta
Use lru_cache for tokenization tests (#36818)
2025-03-28 15:09:35 +01:00
deberta_v2
Use lru_cache for tokenization tests (#36818)
2025-03-28 15:09:35 +01:00
decision_transformer
[CI] Check test if the GenerationTesterMixin inheritance is correct 🐛 🔫 (#36180)
2025-02-21 10:18:20 +00:00
deepseek_v3
[WIP] add deepseek-v3 (#35926)
2025-03-28 15:56:59 +01:00
deformable_detr
chore: fix typos in the tests directory (#36813)
2025-03-21 10:20:05 +01:00
deit
Fix typos in tests (#36547)
2025-03-05 15:04:06 -08:00
depth_anything
chore: fix typos in the tests directory (#36813)
2025-03-21 10:20:05 +01:00
depth_pro
chore: fix typos in the tests directory (#36813)
2025-03-21 10:20:05 +01:00
detr
chore: fix typos in the tests directory (#36813)
2025-03-21 10:20:05 +01:00
diffllama
chore: fix typos in the tests directory (#36813)
2025-03-21 10:20:05 +01:00
dinat
Add common test for torch.export and fix some vision models (#35124)
2025-02-11 11:37:31 +00:00
dinov2
Fix typos in tests (#36547)
2025-03-05 15:04:06 -08:00
dinov2_with_registers
Fix typos in tests (#36547)
2025-03-05 15:04:06 -08:00
distilbert
Use lru_cache for tokenization tests (#36818)
2025-03-28 15:09:35 +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 lru_cache for tokenization tests (#36818)
2025-03-28 15:09:35 +01:00
dpt
chore: fix typos in the tests directory (#36813)
2025-03-21 10:20:05 +01:00
efficientnet
Add common test for torch.export and fix some vision models (#35124)
2025-02-11 11:37:31 +00:00
electra
Use lru_cache for tokenization tests (#36818)
2025-03-28 15:09:35 +01:00
emu3
🔴 VLM: compile compatibility (#35724)
2025-02-14 15:23:49 +01:00
encodec
chore: fix typos in the tests directory (#36813)
2025-03-21 10:20:05 +01:00
encoder_decoder
chore: fix typos in the tests directory (#36813)
2025-03-21 10:20:05 +01:00
ernie
CI: avoid human error, automatically infer generative models (#33212)
2025-02-13 16:27:11 +01:00
esm
Use lru_cache for tokenization tests (#36818)
2025-03-28 15:09:35 +01:00
falcon
chore: fix typos in the tests directory (#36813)
2025-03-21 10:20:05 +01:00
falcon_mamba
chore: fix typos in the tests directory (#36813)
2025-03-21 10:20:05 +01:00
fastspeech2_conformer
Use lru_cache for tokenization tests (#36818)
2025-03-28 15:09:35 +01:00
flaubert
Use lru_cache for tokenization tests (#36818)
2025-03-28 15:09:35 +01:00
flava
Fix typos in tests (#36547)
2025-03-05 15:04:06 -08:00
fnet
Use lru_cache for tokenization tests (#36818)
2025-03-28 15:09:35 +01:00
focalnet
Add common test for torch.export and fix some vision models (#35124)
2025-02-11 11:37:31 +00:00
fsmt
Use lru_cache for tokenization tests (#36818)
2025-03-28 15:09:35 +01:00
funnel
Use lru_cache for tokenization tests (#36818)
2025-03-28 15:09:35 +01:00
fuyu
Update ruff to 0.11.2 (#36962)
2025-03-25 16:00:11 +01:00
gemma
Use lru_cache for tokenization tests (#36818)
2025-03-28 15:09:35 +01:00
gemma2
[generate] ✨ vectorized beam search ✨ (#35802)
2025-03-18 18:39:36 +00:00
gemma3
Deprecate #36741 and map Causal to Conditional (#36917)
2025-03-25 09:13:56 +01:00
git
Fix typos in tests (#36547)
2025-03-05 15:04:06 -08:00
glm
CI: avoid human error, automatically infer generative models (#33212)
2025-02-13 16:27:11 +01:00
glpn
Add common test for torch.export and fix some vision models (#35124)
2025-02-11 11:37:31 +00:00
got_ocr2
Add Got-OCR 2 Fast image processor and refactor slow one (#36185)
2025-03-01 00:56:00 -05:00
gpt2
Use lru_cache for tokenization tests (#36818)
2025-03-28 15:09:35 +01:00
gpt_bigcode
CI: avoid human error, automatically infer generative models (#33212)
2025-02-13 16:27:11 +01:00
gpt_neo
chore: fix typos in the tests directory (#36813)
2025-03-21 10:20:05 +01:00
gpt_neox
Update ruff to 0.11.2 (#36962)
2025-03-25 16:00:11 +01:00
gpt_neox_japanese
Use lru_cache for tokenization tests (#36818)
2025-03-28 15:09:35 +01:00
gpt_sw3
Use lru_cache for tokenization tests (#36818)
2025-03-28 15:09:35 +01:00
gptj
[tests] remove flax-pt equivalence and cross tests (#36283)
2025-02-19 15:13:27 +00:00
granite
chore: fix typos in the tests directory (#36813)
2025-03-21 10:20:05 +01:00
granitemoe
chore: fix typos in the tests directory (#36813)
2025-03-21 10:20:05 +01:00
granitemoeshared
fix typos in the tests directory (#36932)
2025-03-25 10:49:24 +01:00
grounding_dino
[GroundingDino] Fix grounding dino loss 🚨 (#31828)
2025-02-27 19:15:58 +00:00
groupvit
chore: fix typos in the tests directory (#36813)
2025-03-21 10:20:05 +01:00
helium
CI: avoid human error, automatically infer generative models (#33212)
2025-02-13 16:27:11 +01:00
herbert
Use lru_cache for tokenization tests (#36818)
2025-03-28 15:09:35 +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
Set weights_only in torch.load (#36991)
2025-03-27 14:55:50 +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
Fix typos in tests (#36547)
2025-03-05 15:04:06 -08:00
informer
Set weights_only in torch.load (#36991)
2025-03-27 14:55:50 +00:00
instructblip
Simplify keep_in_fp32_modules logic (#36722)
2025-03-21 16:12:59 +01:00
instructblipvideo
Fix typos in tests (#36547)
2025-03-05 15:04:06 -08:00
jamba
CI: avoid human error, automatically infer generative models (#33212)
2025-02-13 16:27:11 +01:00
jetmoe
CI: avoid human error, automatically infer generative models (#33212)
2025-02-13 16:27:11 +01:00
kosmos2
Remove hardcoded slow image processor class in processors supporting fast ones (#36266)
2025-03-12 18:39:25 -04:00
layoutlm
Use lru_cache for tokenization tests (#36818)
2025-03-28 15:09:35 +01:00
layoutlmv2
Use lru_cache for tokenization tests (#36818)
2025-03-28 15:09:35 +01:00
layoutlmv3
Use lru_cache for tokenization tests (#36818)
2025-03-28 15:09:35 +01:00
layoutxlm
Use lru_cache for tokenization tests (#36818)
2025-03-28 15:09:35 +01:00
led
Use lru_cache for tokenization tests (#36818)
2025-03-28 15:09:35 +01:00
levit
[tests] further fix Tester object has no attribute '_testMethodName' (#35781)
2025-01-29 16:05:33 +01:00
lilt
Fix typos in tests (#36547)
2025-03-05 15:04:06 -08:00
llama
Use lru_cache for tokenization tests (#36818)
2025-03-28 15:09:35 +01:00
llava
Fix typos (#36910)
2025-03-24 14:08:29 +00:00
llava_next
Set weights_only in torch.load (#36991)
2025-03-27 14:55:50 +00:00
llava_next_video
chore: fix typos in the tests directory (#36813)
2025-03-21 10:20:05 +01:00
llava_onevision
Fix typos in tests (#36547)
2025-03-05 15:04:06 -08:00
longformer
Use lru_cache for tokenization tests (#36818)
2025-03-28 15:09:35 +01:00
longt5
Fix typos in tests (#36547)
2025-03-05 15:04:06 -08:00
luke
Use lru_cache for tokenization tests (#36818)
2025-03-28 15:09:35 +01:00
lxmert
Use lru_cache for tokenization tests (#36818)
2025-03-28 15:09:35 +01:00
m2m_100
[generate] beam search -- fix output cropping (#37080)
2025-03-28 18:57:51 +01:00
mamba
Remove deprecated batch_size parameter (#37007)
2025-03-27 15:01:56 +00:00
mamba2
CI: avoid human error, automatically infer generative models (#33212)
2025-02-13 16:27:11 +01:00
marian
Use lru_cache for tokenization tests (#36818)
2025-03-28 15:09:35 +01:00
markuplm
Use lru_cache for tokenization tests (#36818)
2025-03-28 15:09:35 +01:00
mask2former
Update ruff to 0.11.2 (#36962)
2025-03-25 16:00:11 +01:00
maskformer
Update ruff to 0.11.2 (#36962)
2025-03-25 16:00:11 +01:00
mbart
Use lru_cache for tokenization tests (#36818)
2025-03-28 15:09:35 +01:00
mbart50
Use lru_cache for tokenization tests (#36818)
2025-03-28 15:09:35 +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 lru_cache for tokenization tests (#36818)
2025-03-28 15:09:35 +01:00
mimi
fix typos in the tests directory (#36932)
2025-03-25 10:49:24 +01:00
mistral
[tests] remove tf/flax tests in /generation (#36235)
2025-02-17 14:59:22 +00:00
mistral3
Deprecate #36741 and map Causal to Conditional (#36917)
2025-03-25 09:13:56 +01:00
mixtral
Fix tensor dtype mismatch (#36985)
2025-03-26 10:37:46 +01:00
mllama
fix: loss computation after embeddings resize - mllama (#36840)
2025-03-21 14:47:59 +01:00
mluke
Use lru_cache for tokenization tests (#36818)
2025-03-28 15:09:35 +01:00
mobilebert
Use lru_cache for tokenization tests (#36818)
2025-03-28 15:09:35 +01:00
mobilenet_v1
Add common test for torch.export and fix some vision models (#35124)
2025-02-11 11:37:31 +00:00
mobilenet_v2
Add common test for torch.export and fix some vision models (#35124)
2025-02-11 11:37:31 +00:00
mobilevit
chore: fix typos in the tests directory (#36813)
2025-03-21 10:20:05 +01:00
mobilevitv2
Add common test for torch.export and fix some vision models (#35124)
2025-02-11 11:37:31 +00:00
modernbert
Support QuestionAnswering Module for ModernBert based models. (#35566)
2025-03-26 21:24:18 +01:00
moonshine
[CI] Check test if the GenerationTesterMixin inheritance is correct 🐛 🔫 (#36180)
2025-02-21 10:18:20 +00:00
moshi
Use lru_cache for tokenization tests (#36818)
2025-03-28 15:09:35 +01:00
mpnet
Use lru_cache for tokenization tests (#36818)
2025-03-28 15:09:35 +01:00
mpt
CI: avoid human error, automatically infer generative models (#33212)
2025-02-13 16:27:11 +01:00
mra
Fix typos in tests (#36547)
2025-03-05 15:04:06 -08:00
mt5
[tests] remove test_export_to_onnx (#36241)
2025-02-17 16:52:44 +00:00
musicgen
chore: fix typos in the tests directory (#36813)
2025-03-21 10:20:05 +01:00
musicgen_melody
chore: fix typos in the tests directory (#36813)
2025-03-21 10:20:05 +01:00
mvp
Use lru_cache for tokenization tests (#36818)
2025-03-28 15:09:35 +01:00
myt5
Use lru_cache for tokenization tests (#36818)
2025-03-28 15:09:35 +01:00
nemotron
CI: avoid human error, automatically infer generative models (#33212)
2025-02-13 16:27:11 +01:00
nllb
Use lru_cache for tokenization tests (#36818)
2025-03-28 15:09:35 +01:00
nllb_moe
Fix typos in tests (#36547)
2025-03-05 15:04:06 -08:00
nougat
Use lru_cache for tokenization tests (#36818)
2025-03-28 15:09:35 +01:00
nystromformer
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
olmo
CI: avoid human error, automatically infer generative models (#33212)
2025-02-13 16:27:11 +01:00
olmo2
CI: avoid human error, automatically infer generative models (#33212)
2025-02-13 16:27:11 +01:00
olmoe
CI: avoid human error, automatically infer generative models (#33212)
2025-02-13 16:27:11 +01:00
omdet_turbo
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
oneformer
chore: fix typos in the tests directory (#36813)
2025-03-21 10:20:05 +01:00
openai
Use lru_cache for tokenization tests (#36818)
2025-03-28 15:09:35 +01:00
opt
[tests] remove tf/flax tests in /generation (#36235)
2025-02-17 14:59:22 +00:00
owlv2
chore: fix typos in the tests directory (#36813)
2025-03-21 10:20:05 +01:00
owlvit
chore: fix typos in the tests directory (#36813)
2025-03-21 10:20:05 +01:00
paligemma
Fix typos (#36910)
2025-03-24 14:08:29 +00:00
paligemma2
Fix typos (#36910)
2025-03-24 14:08:29 +00:00
patchtsmixer
Set weights_only in torch.load (#36991)
2025-03-27 14:55:50 +00:00
patchtst
Set weights_only in torch.load (#36991)
2025-03-27 14:55:50 +00:00
pegasus
Use lru_cache for tokenization tests (#36818)
2025-03-28 15:09:35 +01:00
pegasus_x
[CI] Check test if the GenerationTesterMixin inheritance is correct 🐛 🔫 (#36180)
2025-02-21 10:18:20 +00:00
perceiver
Use lru_cache for tokenization tests (#36818)
2025-03-28 15:09:35 +01:00
persimmon
chore: fix typos in the tests directory (#36813)
2025-03-21 10:20:05 +01:00
phi
chore: fix typos in the tests directory (#36813)
2025-03-21 10:20:05 +01:00
phi3
Export for Phi4-mini (#36780)
2025-03-25 10:46:38 +01:00
phi4_multimodal
Add Phi4 multimodal (#36939)
2025-03-25 09:55:21 +01:00
phimoe
[generate] remove cache v4.47 deprecations (#36212)
2025-02-17 13:55:03 +00:00
phobert
Use lru_cache for tokenization tests (#36818)
2025-03-28 15:09:35 +01:00
pix2struct
Fix typos in tests (#36547)
2025-03-05 15:04:06 -08:00
pixtral
Add Mistral3 (#36790)
2025-03-18 12:04:42 +01:00
plbart
Use lru_cache for tokenization tests (#36818)
2025-03-28 15:09:35 +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
prompt_depth_anything
Add Prompt Depth Anything Model (#35401)
2025-03-20 16:12:44 +00:00
prophetnet
Use lru_cache for tokenization tests (#36818)
2025-03-28 15:09:35 +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
Use lru_cache for tokenization tests (#36818)
2025-03-28 15:09:35 +01:00
qwen2_5_vl
Support return_tensors in audio chat templates (#34601)
2025-03-25 11:08:47 +01:00
qwen2_audio
Support return_tensors in audio chat templates (#34601)
2025-03-25 11:08:47 +01:00
qwen2_moe
Update configuration_qwen2.py (#36735)
2025-03-19 18:15:54 +00:00
qwen2_vl
Support return_tensors in audio chat templates (#34601)
2025-03-25 11:08:47 +01:00
qwen3
Adding Qwen3 and Qwen3MoE (#36878)
2025-03-31 09:50:49 +02:00
qwen3_moe
Adding Qwen3 and Qwen3MoE (#36878)
2025-03-31 09:50:49 +02:00
rag
[tests] make cuda-only tests device-agnostic (#35222)
2024-12-18 10:14:22 +01:00
recurrent_gemma
[Generation, Gemma 3] When passing a custom generation_config, overwrite default values with the model's base generation_config (#36684)
2025-03-15 12:40:09 +00:00
reformer
Use lru_cache for tokenization tests (#36818)
2025-03-28 15:09:35 +01:00
regnet
Add common test for torch.export and fix some vision models (#35124)
2025-02-11 11:37:31 +00:00
rembert
Use lru_cache for tokenization tests (#36818)
2025-03-28 15:09:35 +01:00
resnet
Add common test for torch.export and fix some vision models (#35124)
2025-02-11 11:37:31 +00:00
roberta
Use lru_cache for tokenization tests (#36818)
2025-03-28 15:09:35 +01:00
roberta_prelayernorm
CI: avoid human error, automatically infer generative models (#33212)
2025-02-13 16:27:11 +01:00
roc_bert
Use lru_cache for tokenization tests (#36818)
2025-03-28 15:09:35 +01:00
roformer
Use lru_cache for tokenization tests (#36818)
2025-03-28 15:09:35 +01:00
rt_detr
fix typos in the tests directory (#36932)
2025-03-25 10:49:24 +01:00
rt_detr_v2
chore: fix typos in the tests directory (#36813)
2025-03-21 10:20:05 +01:00
rwkv
CI: avoid human error, automatically infer generative models (#33212)
2025-02-13 16:27:11 +01:00
sam
🚨🚨🚨 Fix sdpa in SAM and refactor relative position embeddings (#36422)
2025-03-17 09:39:52 +00:00
seamless_m4t
Use lru_cache for tokenization tests (#36818)
2025-03-28 15:09:35 +01:00
seamless_m4t_v2
Fix typos in tests (#36547)
2025-03-05 15:04:06 -08:00
segformer
[tests] remove pt_tf equivalence tests (#36253)
2025-02-19 11:55:11 +00:00
seggpt
Add common test for torch.export and fix some vision models (#35124)
2025-02-11 11:37:31 +00:00
sew
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
sew_d
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
shieldgemma2
Support return_tensors in audio chat templates (#34601)
2025-03-25 11:08:47 +01:00
siglip
Use lru_cache for tokenization tests (#36818)
2025-03-28 15:09:35 +01:00
siglip2
Add SigLIP 2 (#36323)
2025-02-21 09:04:19 +00:00
smolvlm
Mark 2 tests as flaky for now (#37038)
2025-03-27 10:59:47 +01:00
speech_encoder_decoder
chore: fix typos in the tests directory (#36813)
2025-03-21 10:20:05 +01:00
speech_to_text
Use lru_cache for tokenization tests (#36818)
2025-03-28 15:09:35 +01:00
speecht5
Use lru_cache for tokenization tests (#36818)
2025-03-28 15:09:35 +01:00
splinter
Use lru_cache for tokenization tests (#36818)
2025-03-28 15:09:35 +01:00
squeezebert
Use lru_cache for tokenization tests (#36818)
2025-03-28 15:09:35 +01:00
stablelm
chore: fix typos in the tests directory (#36813)
2025-03-21 10:20:05 +01:00
starcoder2
CI: avoid human error, automatically infer generative models (#33212)
2025-02-13 16:27:11 +01:00
superglue
CI: avoid human error, automatically infer generative models (#33212)
2025-02-13 16:27:11 +01:00
superpoint
CI: avoid human error, automatically infer generative models (#33212)
2025-02-13 16:27:11 +01:00
swiftformer
Add common test for torch.export and fix some vision models (#35124)
2025-02-11 11:37:31 +00:00
swin
Add common test for torch.export and fix some vision models (#35124)
2025-02-11 11:37:31 +00:00
swin2sr
Fix typos in tests (#36547)
2025-03-05 15:04:06 -08:00
swinv2
Add common test for torch.export and fix some vision models (#35124)
2025-02-11 11:37:31 +00:00
switch_transformers
Fix typos in tests (#36547)
2025-03-05 15:04:06 -08:00
t5
[generate] beam search -- fix output cropping (#37080)
2025-03-28 18:57:51 +01:00
table_transformer
chore: fix typos in the tests directory (#36813)
2025-03-21 10:20:05 +01:00
tapas
Use lru_cache for tokenization tests (#36818)
2025-03-28 15:09:35 +01:00
textnet
Fix tests for vision models (#35654)
2025-02-13 10:28:37 +00:00
time_series_transformer
Set weights_only in torch.load (#36991)
2025-03-27 14:55:50 +00:00
timesformer
Add common test for torch.export and fix some vision models (#35124)
2025-02-11 11:37:31 +00:00
timm_backbone
Less flaky for TimmBackboneModelTest::test_batching_equivalence (#35971)
2025-01-30 16:56:26 +01:00
timm_wrapper
Enable auto task for timm models in pipeline (#35531)
2025-01-08 15:14:17 +00:00
trocr
CI: avoid human error, automatically infer generative models (#33212)
2025-02-13 16:27:11 +01:00
tvp
chore: fix typos in the tests directory (#36813)
2025-03-21 10:20:05 +01:00
udop
Use lru_cache for tokenization tests (#36818)
2025-03-28 15:09:35 +01:00
umt5
Fix typos in tests (#36547)
2025-03-05 15:04:06 -08:00
unispeech
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
unispeech_sat
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
univnet
chore: fix typos in the tests directory (#36813)
2025-03-21 10:20:05 +01:00
upernet
Add common test for torch.export and fix some vision models (#35124)
2025-02-11 11:37:31 +00:00
video_llava
fix typos in the tests directory (#36932)
2025-03-25 10:49:24 +01:00
videomae
Set weights_only in torch.load (#36991)
2025-03-27 14:55:50 +00:00
vilt
Fix typos in tests (#36547)
2025-03-05 15:04:06 -08:00
vipllava
Fix typos (#36910)
2025-03-24 14:08:29 +00:00
vision_encoder_decoder
chore: fix typos in the tests directory (#36813)
2025-03-21 10:20:05 +01:00
vision_text_dual_encoder
[tests] remove flax-pt equivalence and cross tests (#36283)
2025-02-19 15:13:27 +00:00
visual_bert
Fix typos in tests (#36547)
2025-03-05 15:04:06 -08:00
vit
Fix tests for vision models (#35654)
2025-02-13 10:28:37 +00:00
vit_mae
Refactor Attention implementation for ViT-based models (#36545)
2025-03-20 15:15:01 +00:00
vit_msn
[tests] Parameterized test_eager_matches_sdpa_inference (#36650)
2025-03-14 14:41:27 +00:00
vitdet
Fixed VitDet for non-squre Images (#35969)
2025-02-25 19:31:24 +00:00
vitmatte
chore: fix typos in the tests directory (#36813)
2025-03-21 10:20:05 +01:00
vitpose
Add common test for torch.export and fix some vision models (#35124)
2025-02-11 11:37:31 +00:00
vitpose_backbone
skip test_initialization for VitPoseBackboneModelTest for now (#36154)
2025-02-12 18:24:24 +01:00
vits
Use lru_cache for tokenization tests (#36818)
2025-03-28 15:09:35 +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 lru_cache for tokenization tests (#36818)
2025-03-28 15:09:35 +01:00
wav2vec2_bert
Support return_tensors in audio chat templates (#34601)
2025-03-25 11:08:47 +01:00
wav2vec2_conformer
Fix typos in tests (#36547)
2025-03-05 15:04:06 -08:00
wav2vec2_phoneme
Use lru_cache for tokenization tests (#36818)
2025-03-28 15:09:35 +01: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
Use lru_cache for tokenization tests (#36818)
2025-03-28 15:09:35 +01:00
x_clip
chore: fix typos in the tests directory (#36813)
2025-03-21 10:20:05 +01:00
xglm
Use lru_cache for tokenization tests (#36818)
2025-03-28 15:09:35 +01:00
xlm
Use lru_cache for tokenization tests (#36818)
2025-03-28 15:09:35 +01:00
xlm_roberta
Use lru_cache for tokenization tests (#36818)
2025-03-28 15:09:35 +01:00
xlm_roberta_xl
CI: avoid human error, automatically infer generative models (#33212)
2025-02-13 16:27:11 +01:00
xlnet
Use lru_cache for tokenization tests (#36818)
2025-03-28 15:09:35 +01:00
xmod
CI: avoid human error, automatically infer generative models (#33212)
2025-02-13 16:27:11 +01:00
yolos
Add common test for torch.export and fix some vision models (#35124)
2025-02-11 11:37:31 +00:00
yoso
CI: avoid human error, automatically infer generative models (#33212)
2025-02-13 16:27:11 +01:00
zamba
Mark 2 tests as flaky for now (#37038)
2025-03-27 10:59:47 +01:00
zamba2
CI: avoid human error, automatically infer generative models (#33212)
2025-02-13 16:27:11 +01:00
zoedepth
Fix tests for vision models (#35654)
2025-02-13 10:28:37 +00:00
__init__.py
Move test model folders (#17034)
2022-05-03 14:42:02 +02:00