SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity
Files
43d3b1931a7d3cddac9947adcb19bb3b1f8abedb
HuggingFace_transformer/tests/models
History
Yih-Dar 43d3b1931a Update expected output values after #39885 (part 1) (#39990) 
fix

Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
2025-08-07 16:00:28 +02:00
..
aimv2
Rename _supports_flash_attn_2 in examples and tests (#39471)
2025-07-21 14:02:57 +02:00
albert
Deprecate TF + JAX (#38758)
2025-06-11 17:28:06 +01:00
align
Rework add-new-model-like with modular and make test filenames coherent (#39612)
2025-08-04 14:41:09 +02:00
altclip
Rework add-new-model-like with modular and make test filenames coherent (#39612)
2025-08-04 14:41:09 +02:00
arcee
Add Arcee model support (#38621)
2025-06-24 15:05:29 +02:00
aria
Fix aria tests (#39879)
2025-08-05 13:48:47 +02:00
audio_spectrogram_transformer
Add torchcodec in docstrings/tests for datasets 4.0 (#39156)
2025-07-08 17:06:12 +02:00
auto
Enable some ruff checks for performance and readability (#39383)
2025-07-17 13:21:59 +00:00
autoformer
🚨Early-error🚨 config will error out if output_attentions=True and the attn implementation is wrong (#38288)
2025-05-23 17:17:38 +02:00
aya_vision
Rework add-new-model-like with modular and make test filenames coherent (#39612)
2025-08-04 14:41:09 +02:00
bamba
Apply several ruff SIM rules (#37283)
2025-07-29 11:40:34 +00:00
bark
Rework add-new-model-like with modular and make test filenames coherent (#39612)
2025-08-04 14:41:09 +02:00
bart
Deprecate TF + JAX (#38758)
2025-06-11 17:28:06 +01:00
barthez
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
bartpho
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
beit
Update expected output values after #39885 (part 1) (#39990)
2025-08-07 16:00:28 +02:00
bert
[cache] make all classes cache compatible finally (#38635)
2025-07-16 14:00:17 +02:00
bert_generation
Fix typos in strings and comments (#37799)
2025-04-28 11:39:11 +01:00
bert_japanese
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
bertweet
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
big_bird
[cache] make all classes cache compatible finally (#38635)
2025-07-16 14:00:17 +02:00
bigbird_pegasus
[generation] bring back tests on vision models (#38603)
2025-06-06 08:23:15 +00:00
biogpt
Apply several ruff SIM rules (#37283)
2025-07-29 11:40:34 +00:00
bit
Add ImageProcessorFast to BiT processor (#37180)
2025-04-14 17:07:48 +02:00
bitnet
Add Bitnet model (#37742)
2025-04-28 15:08:46 +02:00
blenderbot
Deprecate TF + JAX (#38758)
2025-06-11 17:28:06 +01:00
blenderbot_small
Deprecate TF + JAX (#38758)
2025-06-11 17:28:06 +01:00
blip
Rework add-new-model-like with modular and make test filenames coherent (#39612)
2025-08-04 14:41:09 +02:00
blip_2
Rework add-new-model-like with modular and make test filenames coherent (#39612)
2025-08-04 14:41:09 +02:00
bloom
Apply several ruff SIM rules (#37283)
2025-07-29 11:40:34 +00:00
bridgetower
Rework add-new-model-like with modular and make test filenames coherent (#39612)
2025-08-04 14:41:09 +02:00
bros
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
byt5
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
camembert
Deprecate TF + JAX (#38758)
2025-06-11 17:28:06 +01:00
canine
Skip torchscript tests for 2 models (#38643)
2025-06-06 20:17:37 +02:00
chameleon
Rework add-new-model-like with modular and make test filenames coherent (#39612)
2025-08-04 14:41:09 +02:00
chinese_clip
Rework add-new-model-like with modular and make test filenames coherent (#39612)
2025-08-04 14:41:09 +02:00
clap
Rework add-new-model-like with modular and make test filenames coherent (#39612)
2025-08-04 14:41:09 +02:00
clip
Rework add-new-model-like with modular and make test filenames coherent (#39612)
2025-08-04 14:41:09 +02:00
clipseg
Rework add-new-model-like with modular and make test filenames coherent (#39612)
2025-08-04 14:41:09 +02:00
clvp
Rework add-new-model-like with modular and make test filenames coherent (#39612)
2025-08-04 14:41:09 +02:00
code_llama
remove unhandled parameter (#38145)
2025-06-02 15:57:32 +02:00
codegen
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
cohere
Remove all traces of low_cpu_mem_usage (#38792)
2025-06-12 16:39:33 +02:00
cohere2
[CI] revert device in test_export_static_cache (#39662)
2025-07-25 15:36:12 +00:00
cohere2_vision
Update cohere2 vision test (#39888)
2025-08-04 20:08:18 +02:00
colpali
Rework add-new-model-like with modular and make test filenames coherent (#39612)
2025-08-04 14:41:09 +02:00
colqwen2
Rework add-new-model-like with modular and make test filenames coherent (#39612)
2025-08-04 14:41:09 +02:00
conditional_detr
Update expected output values after #39885 (part 1) (#39990)
2025-08-07 16:00:28 +02:00
convbert
Deprecate TF + JAX (#38758)
2025-06-11 17:28:06 +01:00
convnext
Update expected values (after switching to A10) (#39157)
2025-07-01 20:54:31 +02:00
convnextv2
Deprecate TF + JAX (#38758)
2025-06-11 17:28:06 +01:00
cpm
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
cpmant
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
csm
Rework add-new-model-like with modular and make test filenames coherent (#39612)
2025-08-04 14:41:09 +02:00
ctrl
Deprecate TF + JAX (#38758)
2025-06-11 17:28:06 +01:00
cvt
Update expected values (after switching to A10) (#39157)
2025-07-01 20:54:31 +02:00
d_fine
Update expected output values after #39885 (part 1) (#39990)
2025-08-07 16:00:28 +02:00
dab_detr
Update expected output values after #39885 (part 1) (#39990)
2025-08-07 16:00:28 +02:00
dac
Apply several ruff SIM rules (#37283)
2025-07-29 11:40:34 +00:00
data2vec
Add torchcodec in docstrings/tests for datasets 4.0 (#39156)
2025-07-08 17:06:12 +02:00
dbrx
More robust tied weight test (#39681)
2025-07-25 22:03:21 +02:00
deberta
Deprecate TF + JAX (#38758)
2025-06-11 17:28:06 +01:00
deberta_v2
Deprecate TF + JAX (#38758)
2025-06-11 17:28:06 +01:00
decision_transformer
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
deepseek_v2
[cache refactor] Move all the caching logic to a per-layer approach (#39106)
2025-07-22 16:10:25 +02:00
deepseek_v3
[cache refactor] Move all the caching logic to a per-layer approach (#39106)
2025-07-22 16:10:25 +02:00
deepseek_vl
Rework add-new-model-like with modular and make test filenames coherent (#39612)
2025-08-04 14:41:09 +02:00
deepseek_vl_hybrid
Rework add-new-model-like with modular and make test filenames coherent (#39612)
2025-08-04 14:41:09 +02:00
deformable_detr
Update expected output values after #39885 (part 1) (#39990)
2025-08-07 16:00:28 +02:00
deit
Deprecate TF + JAX (#38758)
2025-06-11 17:28:06 +01:00
depth_anything
Simplify conditional code (#39781)
2025-07-30 12:32:10 +00:00
depth_pro
Update expected values (after switching to A10) - part 7 (#39218)
2025-07-04 12:48:10 +02:00
detr
Update expected output values after #39885 (part 1) (#39990)
2025-08-07 16:00:28 +02:00
dia
Rework add-new-model-like with modular and make test filenames coherent (#39612)
2025-08-04 14:41:09 +02:00
diffllama
Remove all traces of low_cpu_mem_usage (#38792)
2025-06-12 16:39:33 +02:00
dinat
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
dinov2
Deprecate TF + JAX (#38758)
2025-06-11 17:28:06 +01:00
dinov2_with_registers
Fix more flaky test_initialization (#38932)
2025-06-20 17:28:32 +02:00
distilbert
Deprecate TF + JAX (#38758)
2025-06-11 17:28:06 +01:00
dit
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
doge
Add Doge model (#35891)
2025-07-08 11:44:29 +02:00
donut
Rework add-new-model-like with modular and make test filenames coherent (#39612)
2025-08-04 14:41:09 +02:00
dots1
fix dots1 tests (#39088)
2025-06-27 16:54:11 +02:00
dpr
Deprecate TF + JAX (#38758)
2025-06-11 17:28:06 +01:00
dpt
Update expected output values after #39885 (part 1) (#39990)
2025-08-07 16:00:28 +02:00
efficientloftr
[configuration] remove redundant classmethod (#38812)
2025-07-28 10:38:48 +00:00
efficientnet
Add EfficientNet Image PreProcessor (#37055)
2025-04-16 21:59:24 +02:00
electra
Deprecate TF + JAX (#38758)
2025-06-11 17:28:06 +01:00
emu3
Rework add-new-model-like with modular and make test filenames coherent (#39612)
2025-08-04 14:41:09 +02:00
encodec
Apply several ruff SIM rules (#37283)
2025-07-29 11:40:34 +00:00
encoder_decoder
Deprecate TF + JAX (#38758)
2025-06-11 17:28:06 +01:00
eomt
🚨🚨🚨 [eomt] make EoMT compatible with pipeline (#39122)
2025-07-02 12:25:26 +01:00
ernie
Remove old code for PyTorch, Accelerator and tokenizers (#37234)
2025-04-10 20:54:21 +02:00
ernie4_5
[Ernie 4.5] Post merge adaptations (#39664)
2025-07-25 17:36:18 +02:00
ernie4_5_moe
[Ernie 4.5] Post merge adaptations (#39664)
2025-07-25 17:36:18 +02:00
esm
Rename _supports_flash_attn_2 in examples and tests (#39471)
2025-07-21 14:02:57 +02:00
evolla
Rework add-new-model-like with modular and make test filenames coherent (#39612)
2025-08-04 14:41:09 +02:00
exaone4
Add EXAONE 4.0 model (#39129)
2025-07-25 19:58:28 +02:00
falcon
🚨 🚨 Inherited CausalLM Tests (#37590)
2025-05-23 18:29:31 +01:00
falcon_h1
Apply several ruff SIM rules (#37283)
2025-07-29 11:40:34 +00:00
falcon_mamba
Apply several ruff SIM rules (#37283)
2025-07-29 11:40:34 +00:00
fastspeech2_conformer
Update expected output values after #39885 (part 1) (#39990)
2025-08-07 16:00:28 +02:00
flaubert
Deprecate TF + JAX (#38758)
2025-06-11 17:28:06 +01:00
flava
Rework add-new-model-like with modular and make test filenames coherent (#39612)
2025-08-04 14:41:09 +02:00
fnet
🚨 rm already deprecated pad_to_max_length arg (#37617)
2025-05-01 15:21:55 +02:00
focalnet
Update expected values (after switching to A10) - part 4 (#39189)
2025-07-03 15:13:06 +02:00
fsmt
Fix fsmt tests (#38904)
2025-06-19 10:56:34 +02:00
funnel
Deprecate TF + JAX (#38758)
2025-06-11 17:28:06 +01:00
fuyu
Rework add-new-model-like with modular and make test filenames coherent (#39612)
2025-08-04 14:41:09 +02:00
gemma
Support input_embeds in torch exportable decoders (#39836)
2025-08-07 08:51:31 +00:00
gemma2
Support input_embeds in torch exportable decoders (#39836)
2025-08-07 08:51:31 +00:00
gemma3
Support input_embeds in torch exportable decoders (#39836)
2025-08-07 08:51:31 +00:00
gemma3n
Fix gemma3n feature extractor's incorrect squeeze (#39919)
2025-08-07 18:34:28 +08:00
git
Rework add-new-model-like with modular and make test filenames coherent (#39612)
2025-08-04 14:41:09 +02:00
glm
Update expected values (after switching to A10) - part 5 (#39205)
2025-07-03 19:56:02 +02:00
glm4
enable misc test cases on XPU (#38852)
2025-06-18 09:20:49 +02:00
glm4_moe
skip Glm4MoeModelTest::test_torch_compile_for_training (#39670)
2025-07-28 16:30:40 +02:00
glm4v
fix Glm4v batch videos forward (#39172)
2025-07-10 10:44:28 +02:00
glpn
Update expected values (after switching to A10) - part 4 (#39189)
2025-07-03 15:13:06 +02:00
got_ocr2
Rework add-new-model-like with modular and make test filenames coherent (#39612)
2025-08-04 14:41:09 +02:00
gpt2
Apply several ruff SIM rules (#37283)
2025-07-29 11:40:34 +00:00
gpt_bigcode
Apply several ruff SIM rules (#37283)
2025-07-29 11:40:34 +00:00
gpt_neo
Deprecate TF + JAX (#38758)
2025-06-11 17:28:06 +01:00
gpt_neox
[cache refactor] Move all the caching logic to a per-layer approach (#39106)
2025-07-22 16:10:25 +02:00
gpt_neox_japanese
Remove old code for PyTorch, Accelerator and tokenizers (#37234)
2025-04-10 20:54:21 +02:00
gpt_oss
Add GPT OSS model from OpenAI (#39923)
2025-08-05 18:02:18 +02:00
gpt_sw3
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
gptj
Deprecate TF + JAX (#38758)
2025-06-11 17:28:06 +01:00
granite
switch to device agnostic device calling for test cases (#38247)
2025-05-26 10:18:53 +02:00
granite_speech
Rework add-new-model-like with modular and make test filenames coherent (#39612)
2025-08-04 14:41:09 +02:00
granitemoe
switch to device agnostic device calling for test cases (#38247)
2025-05-26 10:18:53 +02:00
granitemoehybrid
switch to device agnostic device calling for test cases (#38247)
2025-05-26 10:18:53 +02:00
granitemoeshared
switch to device agnostic device calling for test cases (#38247)
2025-05-26 10:18:53 +02:00
grounding_dino
Update expected output values after #39885 (part 1) (#39990)
2025-08-07 16:00:28 +02:00
groupvit
Apply several ruff SIM rules (#37283)
2025-07-29 11:40:34 +00:00
helium
Update expected values (after switching to A10) - part 5 (#39205)
2025-07-03 19:56:02 +02:00
herbert
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
hgnet_v2
Add D-FINE Model into Transformers (#36261)
2025-04-29 12:17:55 +01:00
hiera
Simplify conditional code (#39781)
2025-07-30 12:32:10 +00:00
hubert
Fix bad tensor shape in failing Hubert test. (#39502)
2025-07-21 12:25:52 +01:00
ibert
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
idefics
Rework add-new-model-like with modular and make test filenames coherent (#39612)
2025-08-04 14:41:09 +02:00
idefics2
Rework add-new-model-like with modular and make test filenames coherent (#39612)
2025-08-04 14:41:09 +02:00
idefics3
Rework add-new-model-like with modular and make test filenames coherent (#39612)
2025-08-04 14:41:09 +02:00
ijepa
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
imagegpt
[test] update test_past_key_values_format (#37614)
2025-04-22 11:07:34 +01:00
informer
🚨Early-error🚨 config will error out if output_attentions=True and the attn implementation is wrong (#38288)
2025-05-23 17:17:38 +02:00
instructblip
Rework add-new-model-like with modular and make test filenames coherent (#39612)
2025-08-04 14:41:09 +02:00
instructblipvideo
Rework add-new-model-like with modular and make test filenames coherent (#39612)
2025-08-04 14:41:09 +02:00
internvl
Various test fixes for AMD (#39978)
2025-08-07 10:57:04 +02:00
jamba
Expectation fixes and added AMD expectations (#38729)
2025-06-13 16:14:58 +02:00
janus
Rework add-new-model-like with modular and make test filenames coherent (#39612)
2025-08-04 14:41:09 +02:00
jetmoe
🚨 🚨 Inherited CausalLM Tests (#37590)
2025-05-23 18:29:31 +01:00
kosmos2
Rework add-new-model-like with modular and make test filenames coherent (#39612)
2025-08-04 14:41:09 +02:00
kyutai_speech_to_text
fix: providing a tensor to cache_position in model.generate kwargs always crashes because of boolean test (#39300)
2025-07-30 17:30:28 +00:00
layoutlm
🚨 Don't use cache in non-generative models (#38751)
2025-07-01 09:08:21 +00:00
layoutlmv2
Rework add-new-model-like with modular and make test filenames coherent (#39612)
2025-08-04 14:41:09 +02:00
layoutlmv3
Rework add-new-model-like with modular and make test filenames coherent (#39612)
2025-08-04 14:41:09 +02:00
layoutxlm
Rework add-new-model-like with modular and make test filenames coherent (#39612)
2025-08-04 14:41:09 +02:00
led
Deprecate TF + JAX (#38758)
2025-06-11 17:28:06 +01:00
levit
Update expected output values after #39885 (part 1) (#39990)
2025-08-07 16:00:28 +02:00
lfm2
Fix Lfm2 and common tests (#39398)
2025-07-14 12:02:59 +02:00
lightglue
Update expected values (after switching to A10) - part 4 (#39189)
2025-07-03 15:13:06 +02:00
lilt
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
llama
Various test fixes for AMD (#39978)
2025-08-07 10:57:04 +02:00
llama4
Rework add-new-model-like with modular and make test filenames coherent (#39612)
2025-08-04 14:41:09 +02:00
llava
Various test fixes for AMD (#39978)
2025-08-07 10:57:04 +02:00
llava_next
Rework add-new-model-like with modular and make test filenames coherent (#39612)
2025-08-04 14:41:09 +02:00
llava_next_video
Rework add-new-model-like with modular and make test filenames coherent (#39612)
2025-08-04 14:41:09 +02:00
llava_onevision
Rework add-new-model-like with modular and make test filenames coherent (#39612)
2025-08-04 14:41:09 +02:00
longformer
Deprecate TF + JAX (#38758)
2025-06-11 17:28:06 +01:00
longt5
Deprecate TF + JAX (#38758)
2025-06-11 17:28:06 +01:00
luke
Apply several ruff SIM rules (#37283)
2025-07-29 11:40:34 +00:00
lxmert
Remove all traces of low_cpu_mem_usage (#38792)
2025-06-12 16:39:33 +02:00
m2m_100
🔴🔴🔴 [Attention] Refactor Attention Interface for Bart-based Models (#38108)
2025-05-22 17:12:58 +02:00
mamba
Apply several ruff SIM rules (#37283)
2025-07-29 11:40:34 +00:00
mamba2
More robust tied weight test (#39681)
2025-07-25 22:03:21 +02:00
marian
Remove all traces of low_cpu_mem_usage (#38792)
2025-06-12 16:39:33 +02:00
markuplm
Rework add-new-model-like with modular and make test filenames coherent (#39612)
2025-08-04 14:41:09 +02:00
mask2former
Update expected output values after #39885 (part 1) (#39990)
2025-08-07 16:00:28 +02:00
maskformer
Update expected output values after #39885 (part 1) (#39990)
2025-08-07 16:00:28 +02:00
mbart
Deprecate TF + JAX (#38758)
2025-06-11 17:28:06 +01:00
mbart50
Use lru_cache for tokenization tests (#36818)
2025-03-28 15:09:35 +01:00
megatron_bert
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
megatron_gpt2
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
mgp_str
Rework add-new-model-like with modular and make test filenames coherent (#39612)
2025-08-04 14:41:09 +02:00
mimi
Apply several ruff SIM rules (#37283)
2025-07-29 11:40:34 +00:00
minimax
[attention] fix test for packed padfree masking (#39582)
2025-07-25 07:44:52 +00:00
mistral
fix mistral and mistral3 tests (#38978)
2025-06-23 17:07:18 +02:00
mistral3
Various test fixes for AMD (#39978)
2025-08-07 10:57:04 +02:00
mixtral
Update expected values (after switching to A10) - part 4 (#39189)
2025-07-03 15:13:06 +02:00
mlcd
Add MLCD model (#36182)
2025-04-15 11:33:09 +01:00
mllama
Rework add-new-model-like with modular and make test filenames coherent (#39612)
2025-08-04 14:41:09 +02:00
mluke
Fix typos in strings and comments (#37799)
2025-04-28 11:39:11 +01:00
mm_grounding_dino
Add MM Grounding DINO (#37925)
2025-08-01 15:43:23 +01:00
mobilebert
Deprecate TF + JAX (#38758)
2025-06-11 17:28:06 +01:00
mobilenet_v1
Update expected output values after #39885 (part 1) (#39990)
2025-08-07 16:00:28 +02:00
mobilenet_v2
Update expected output values after #39885 (part 1) (#39990)
2025-08-07 16:00:28 +02:00
mobilevit
Update expected output values after #39885 (part 1) (#39990)
2025-08-07 16:00:28 +02:00
mobilevitv2
Update expected output values after #39885 (part 1) (#39990)
2025-08-07 16:00:28 +02:00
modernbert
Add support for ModernBertForMultipleChoice (#39232)
2025-08-04 20:45:43 +02:00
modernbert_decoder
Add ModernBERT Decoder Models - ModernBERT, but trained with CLM! (#38967)
2025-07-15 10:40:41 +02:00
moonshine
Update expected output values after #39885 (part 1) (#39990)
2025-08-07 16:00:28 +02:00
moshi
Apply several ruff SIM rules (#37283)
2025-07-29 11:40:34 +00:00
mpnet
Deprecate TF + JAX (#38758)
2025-06-11 17:28:06 +01:00
mpt
Apply several ruff SIM rules (#37283)
2025-07-29 11:40:34 +00:00
mra
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
mt5
Deprecate TF + JAX (#38758)
2025-06-11 17:28:06 +01:00
musicgen
Rework add-new-model-like with modular and make test filenames coherent (#39612)
2025-08-04 14:41:09 +02:00
musicgen_melody
Rework add-new-model-like with modular and make test filenames coherent (#39612)
2025-08-04 14:41:09 +02:00
mvp
[cache] make all classes cache compatible finally (#38635)
2025-07-16 14:00:17 +02:00
myt5
🚨 🚨 Setup -> setupclass conversion (#37282)
2025-04-08 17:15:37 +01:00
nemotron
switch to device agnostic device calling for test cases (#38247)
2025-05-26 10:18:53 +02:00
nllb
Use lru_cache for tokenization tests (#36818)
2025-03-28 15:09:35 +01:00
nllb_moe
Fix typos in strings and comments (#37799)
2025-04-28 11:39:11 +01:00
nougat
add fast image processor nougat (#37661)
2025-06-27 14:39:43 +00:00
nystromformer
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
olmo
Support input_embeds in torch exportable decoders (#39836)
2025-08-07 08:51:31 +00:00
olmo2
[CI] revert device in test_export_static_cache (#39662)
2025-07-25 15:36:12 +00:00
olmoe
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
omdet_turbo
Rework add-new-model-like with modular and make test filenames coherent (#39612)
2025-08-04 14:41:09 +02:00
oneformer
Update expected output values after #39885 (part 1) (#39990)
2025-08-07 16:00:28 +02:00
openai
Deprecate TF + JAX (#38758)
2025-06-11 17:28:06 +01:00
opt
Deprecate TF + JAX (#38758)
2025-06-11 17:28:06 +01:00
owlv2
Rework add-new-model-like with modular and make test filenames coherent (#39612)
2025-08-04 14:41:09 +02:00
owlvit
Rework add-new-model-like with modular and make test filenames coherent (#39612)
2025-08-04 14:41:09 +02:00
paligemma
Rework add-new-model-like with modular and make test filenames coherent (#39612)
2025-08-04 14:41:09 +02:00
paligemma2
[attention] fix test for packed padfree masking (#39582)
2025-07-25 07:44:52 +00:00
patchtsmixer
🔴🔴🔴 [Attention] Refactor Attention Interface for Bart-based Models (#38108)
2025-05-22 17:12:58 +02:00
patchtst
Force torch>=2.6 with torch.load to avoid vulnerability issue (#37785)
2025-04-25 16:57:09 +02:00
pegasus
Deprecate TF + JAX (#38758)
2025-06-11 17:28:06 +01:00
pegasus_x
🚨Early-error🚨 config will error out if output_attentions=True and the attn implementation is wrong (#38288)
2025-05-23 17:17:38 +02:00
perceiver
Apply several ruff SIM rules (#37283)
2025-07-29 11:40:34 +00:00
perception_lm
Rework add-new-model-like with modular and make test filenames coherent (#39612)
2025-08-04 14:41:09 +02:00
persimmon
[attention] fix test for packed padfree masking (#39582)
2025-07-25 07:44:52 +00:00
phi
🚨 🚨 Inherited CausalLM Tests (#37590)
2025-05-23 18:29:31 +01:00
phi3
Support input_embeds in torch exportable decoders (#39836)
2025-08-07 08:51:31 +00:00
phi4_multimodal
Rework add-new-model-like with modular and make test filenames coherent (#39612)
2025-08-04 14:41:09 +02:00
phimoe
Fix MoE gradient test (#38438)
2025-05-28 16:44:20 +01:00
phobert
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
pix2struct
Rework add-new-model-like with modular and make test filenames coherent (#39612)
2025-08-04 14:41:09 +02:00
pixtral
Rework add-new-model-like with modular and make test filenames coherent (#39612)
2025-08-04 14:41:09 +02:00
plbart
🔴🔴🔴 [Attention] Refactor Attention Interface for Bart-based Models (#38108)
2025-05-22 17:12:58 +02:00
poolformer
Update expected output values after #39885 (part 1) (#39990)
2025-08-07 16:00:28 +02:00
pop2piano
Rework add-new-model-like with modular and make test filenames coherent (#39612)
2025-08-04 14:41:09 +02:00
prompt_depth_anything
Skip some export tests on torch 2.7 (#38677)
2025-06-12 12:47:15 +02:00
prophetnet
[cache] make all classes cache compatible finally (#38635)
2025-07-16 14:00:17 +02:00
pvt
Update expected output values after #39885 (part 1) (#39990)
2025-08-07 16:00:28 +02:00
pvt_v2
Update expected values (after switching to A10) - part 3 (#39179)
2025-07-02 22:48:30 +02:00
qwen2
Support input_embeds in torch exportable decoders (#39836)
2025-08-07 08:51:31 +00:00
qwen2_5_omni
Replace video_fps with fps in tests (#39898)
2025-08-05 10:39:55 +00:00
qwen2_5_vl
Replace video_fps with fps in tests (#39898)
2025-08-05 10:39:55 +00:00
qwen2_audio
Rework add-new-model-like with modular and make test filenames coherent (#39612)
2025-08-04 14:41:09 +02:00
qwen2_moe
Fix MoE gradient test (#38438)
2025-05-28 16:44:20 +01:00
qwen2_vl
Replace video_fps with fps in tests (#39898)
2025-08-05 10:39:55 +00:00
qwen3
Support input_embeds in torch exportable decoders (#39836)
2025-08-07 08:51:31 +00:00
qwen3_moe
Fix qwen3_moe tests (#38865)
2025-06-18 14:36:03 +02:00
rag
Fix rag (#38585)
2025-06-23 17:42:46 +02:00
recurrent_gemma
adjust input and output texts for test_modeling_recurrent_gemma.py (#39190)
2025-07-07 15:13:25 +02:00
reformer
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
regnet
Update expected output values after #39885 (part 1) (#39990)
2025-08-07 16:00:28 +02:00
rembert
Deprecate TF + JAX (#38758)
2025-06-11 17:28:06 +01:00
resnet
Update expected output values after #39885 (part 1) (#39990)
2025-08-07 16:00:28 +02:00
roberta
Deprecate TF + JAX (#38758)
2025-06-11 17:28:06 +01:00
roberta_prelayernorm
Deprecate TF + JAX (#38758)
2025-06-11 17:28:06 +01:00
roc_bert
Remove old code for PyTorch, Accelerator and tokenizers (#37234)
2025-04-10 20:54:21 +02:00
roformer
Deprecate TF + JAX (#38758)
2025-06-11 17:28:06 +01:00
rt_detr
Update expected output values after #39885 (part 1) (#39990)
2025-08-07 16:00:28 +02:00
rt_detr_v2
Update expected output values after #39885 (part 1) (#39990)
2025-08-07 16:00:28 +02:00
rwkv
Apply several ruff SIM rules (#37283)
2025-07-29 11:40:34 +00:00
sam
Update expected output values after #39885 (part 1) (#39990)
2025-08-07 16:00:28 +02:00
sam_hq
Rework add-new-model-like with modular and make test filenames coherent (#39612)
2025-08-04 14:41:09 +02:00
seamless_m4t
Rework add-new-model-like with modular and make test filenames coherent (#39612)
2025-08-04 14:41:09 +02:00
seamless_m4t_v2
[seamless_m4t] Skip some tests when speech is not available (#38430)
2025-06-02 09:17:28 +00:00
segformer
Add Fast Segformer Processor (#37024)
2025-07-28 19:22:32 +00:00
seggpt
Update expected values (after switching to A10) - part 3 (#39179)
2025-07-02 22:48:30 +02:00
sew
Add torchcodec in docstrings/tests for datasets 4.0 (#39156)
2025-07-08 17:06:12 +02:00
sew_d
Add torchcodec in docstrings/tests for datasets 4.0 (#39156)
2025-07-08 17:06:12 +02:00
shieldgemma2
[chat template] return assistant mask in processors (#38545)
2025-07-18 12:23:20 +00:00
siglip
Apply several ruff SIM rules (#37283)
2025-07-29 11:40:34 +00:00
siglip2
[configuration] remove redundant classmethod (#38812)
2025-07-28 10:38:48 +00:00
smollm3
[CI] revert device in test_export_static_cache (#39662)
2025-07-25 15:36:12 +00:00
smolvlm
Export SmolvLM (#39614)
2025-08-05 16:20:23 +02:00
speech_encoder_decoder
[attn_implementation] remove recursive, allows custom kernels with wrappers (#39823)
2025-08-01 12:18:28 +02:00
speech_to_text
Rework add-new-model-like with modular and make test filenames coherent (#39612)
2025-08-04 14:41:09 +02:00
speecht5
Rework add-new-model-like with modular and make test filenames coherent (#39612)
2025-08-04 14:41:09 +02:00
splinter
🚨 Don't use cache in non-generative models (#38751)
2025-07-01 09:08:21 +00:00
squeezebert
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
stablelm
🚨 🚨 Inherited CausalLM Tests (#37590)
2025-05-23 18:29:31 +01:00
starcoder2
Update expected values (after switching to A10) - part 5 (#39205)
2025-07-03 19:56:02 +02:00
superglue
[superglue] Fixed the way batch mask was applied to the scores before match assignment computation (#39968)
2025-08-07 09:49:39 +01:00
superpoint
Superpoint fast image processor (#37804)
2025-07-28 18:15:06 +00:00
swiftformer
Fix missing initializations for models created in 2023 (#39239)
2025-07-21 14:43:52 +02:00
swin
Simplify conditional code (#39781)
2025-07-30 12:32:10 +00:00
swin2sr
Update expected values (after switching to A10) - part 3 (#39179)
2025-07-02 22:48:30 +02:00
swinv2
Simplify conditional code (#39781)
2025-07-30 12:32:10 +00:00
switch_transformers
Update expected values (after switching to A10) - part 3 (#39179)
2025-07-02 22:48:30 +02:00
t5
Fix convert_and_export_with_cache failures for GPU models (#38976)
2025-07-17 13:12:32 +00:00
t5gemma
fix: providing a tensor to cache_position in model.generate kwargs always crashes because of boolean test (#39300)
2025-07-30 17:30:28 +00:00
table_transformer
🚨Early-error🚨 config will error out if output_attentions=True and the attn implementation is wrong (#38288)
2025-05-23 17:17:38 +02:00
tapas
Apply several ruff SIM rules (#37283)
2025-07-29 11:40:34 +00:00
textnet
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
time_series_transformer
🚨Early-error🚨 config will error out if output_attentions=True and the attn implementation is wrong (#38288)
2025-05-23 17:17:38 +02:00
timesfm
Remove dead protected imports (#38980)
2025-06-23 13:44:50 +02:00
timesformer
Update expected output values after #39885 (part 1) (#39990)
2025-08-07 16:00:28 +02:00
timm_backbone
More robust tied weight test (#39681)
2025-07-25 22:03:21 +02:00
timm_wrapper
[timm_wrapper] add support for gradient checkpointing (#39287)
2025-07-22 11:07:52 +00:00
trocr
Rework add-new-model-like with modular and make test filenames coherent (#39612)
2025-08-04 14:41:09 +02:00
tvp
Add Optional to remaining types (#37808)
2025-04-28 14:20:45 +01:00
udop
Rework add-new-model-like with modular and make test filenames coherent (#39612)
2025-08-04 14:41:09 +02:00
umt5
[generation] bring back tests on vision models (#38603)
2025-06-06 08:23:15 +00:00
unispeech
Add torchcodec in docstrings/tests for datasets 4.0 (#39156)
2025-07-08 17:06:12 +02:00
unispeech_sat
Add torchcodec in docstrings/tests for datasets 4.0 (#39156)
2025-07-08 17:06:12 +02:00
univnet
Add torchcodec in docstrings/tests for datasets 4.0 (#39156)
2025-07-08 17:06:12 +02:00
upernet
Update expected output values after #39885 (part 1) (#39990)
2025-08-07 16:00:28 +02:00
video_llava
[VLMs] support passing embeds along with pixels (#38467)
2025-07-01 11:33:20 +00:00
videomae
Rename _supports_flash_attn_2 in examples and tests (#39471)
2025-07-21 14:02:57 +02:00
vilt
Remove script datasets in tests (#38940)
2025-06-25 14:31:20 +00:00
vipllava
[VLMs] support passing embeds along with pixels (#38467)
2025-07-01 11:33:20 +00:00
vision_encoder_decoder
[attn_implementation] remove recursive, allows custom kernels with wrappers (#39823)
2025-08-01 12:18:28 +02:00
vision_text_dual_encoder
Rework add-new-model-like with modular and make test filenames coherent (#39612)
2025-08-04 14:41:09 +02:00
visual_bert
Apply several ruff SIM rules (#37283)
2025-07-29 11:40:34 +00:00
vit
Deprecate TF + JAX (#38758)
2025-06-11 17:28:06 +01:00
vit_mae
Rename _supports_flash_attn_2 in examples and tests (#39471)
2025-07-21 14:02:57 +02:00
vit_msn
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
vitdet
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
vitmatte
Simplify conditional code (#39781)
2025-07-30 12:32:10 +00:00
vitpose
Simplify conditional code (#39781)
2025-07-30 12:32:10 +00:00
vitpose_backbone
Update expected values (after switching to A10) - part 4 (#39189)
2025-07-03 15:13:06 +02:00
vits
Apply several ruff SIM rules (#37283)
2025-07-29 11:40:34 +00:00
vivit
Update expected output values after #39885 (part 1) (#39990)
2025-08-07 16:00:28 +02:00
vjepa2
Refactor the way we handle outputs for new llamas and new models (#39120)
2025-07-05 11:34:28 +02:00
voxtral
fix(voxtral): correct typo in apply_transcription_request (#39572)
2025-07-25 12:09:44 +00:00
wav2vec2
Rework add-new-model-like with modular and make test filenames coherent (#39612)
2025-08-04 14:41:09 +02:00
wav2vec2_bert
Rework add-new-model-like with modular and make test filenames coherent (#39612)
2025-08-04 14:41:09 +02:00
wav2vec2_conformer
Update expected values (after switching to A10) - part 4 (#39189)
2025-07-03 15:13:06 +02:00
wav2vec2_phoneme
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
wav2vec2_with_lm
Rework add-new-model-like with modular and make test filenames coherent (#39612)
2025-08-04 14:41:09 +02:00
wavlm
Remove script datasets in tests (#38940)
2025-06-25 14:31:20 +00:00
whisper
Rework add-new-model-like with modular and make test filenames coherent (#39612)
2025-08-04 14:41:09 +02:00
x_clip
Update expected output values after #39885 (part 1) (#39990)
2025-08-07 16:00:28 +02:00
xglm
Apply several ruff SIM rules (#37283)
2025-07-29 11:40:34 +00:00
xlm
Deprecate TF + JAX (#38758)
2025-06-11 17:28:06 +01:00
xlm_roberta
Deprecate TF + JAX (#38758)
2025-06-11 17:28:06 +01:00
xlm_roberta_xl
Remove old code for PyTorch, Accelerator and tokenizers (#37234)
2025-04-10 20:54:21 +02:00
xlnet
Deprecate TF + JAX (#38758)
2025-06-11 17:28:06 +01:00
xlstm
Fix Evolla and xLSTM tests (#39769)
2025-07-30 09:51:55 +02:00
xmod
Remove old code for PyTorch, Accelerator and tokenizers (#37234)
2025-04-10 20:54:21 +02:00
yolos
🚨Early-error🚨 config will error out if output_attentions=True and the attn implementation is wrong (#38288)
2025-05-23 17:17:38 +02:00
yoso
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
zamba
Remove all traces of low_cpu_mem_usage (#38792)
2025-06-12 16:39:33 +02:00
zamba2
fix UT failures on XPU w/ stock PyTorch 2.7 & 2.8 (#39116)
2025-06-30 11:49:03 +02:00
zoedepth
Simplify conditional code (#39781)
2025-07-30 12:32:10 +00:00
__init__.py
…