SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity
Files
d69945e5fced637e77cf7af5e4955cb897bc298c
HuggingFace_transformer/tests/models
History
Peter St. John d69945e5fc [ESM] Add flash-attention-2 backend for ESM-2 (#38023) 
* Add flash-attention-2 backend for ESM-2

Signed-off-by: Peter St. John <pstjohn@nvidia.com>

* update extended_attention_mask for fa2

Signed-off-by: Peter St. John <pstjohn@nvidia.com>

* add test_flash_attn_2_equivalence test

Signed-off-by: Peter St. John <pstjohn@nvidia.com>

---------

Signed-off-by: Peter St. John <pstjohn@nvidia.com>
2025-05-16 14:11:56 +01:00
..
albert
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
align
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
altclip
🚨 🚨 Setup -> setupclass conversion (#37282)
2025-04-08 17:15:37 +01:00
aria
🔴 [VLM] Add base model without head (#37033)
2025-05-07 17:47:51 +02:00
audio_spectrogram_transformer
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
auto
🔴 Video processors as a separate class (#35206)
2025-05-12 11:55:51 +02:00
autoformer
Force torch>=2.6 with torch.load to avoid vulnerability issue (#37785)
2025-04-25 16:57:09 +02:00
aya_vision
update require_read_token (#38093)
2025-05-13 12:07:07 +02:00
bamba
Add GraniteMoeHybrid support for 4.0 (#37658)
2025-05-06 06:47:43 +02:00
bark
enable cpu offloading for Bark on xpu (#37599)
2025-04-23 11:37:15 +02:00
bart
Bart: new cache format (#35314)
2025-05-16 13:26:54 +02: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
[Fast Processor] BEiT (#37005)
2025-05-06 17:40:28 -04:00
bert
Fix typos in strings and comments (#37784)
2025-04-25 13:47:25 +01: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
Fix typos in strings and comments (#37799)
2025-04-28 11:39:11 +01:00
bigbird_pegasus
Bart: new cache format (#35314)
2025-05-16 13:26:54 +02:00
biogpt
Bart: new cache format (#35314)
2025-05-16 13:26:54 +02: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
Bart: new cache format (#35314)
2025-05-16 13:26:54 +02:00
blenderbot_small
Bart: new cache format (#35314)
2025-05-16 13:26:54 +02:00
blip
🚨 🚨 Setup -> setupclass conversion (#37282)
2025-04-08 17:15:37 +01:00
blip_2
[VLMs] support attention backends (#37576)
2025-05-08 18:18:54 +02:00
bloom
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
bridgetower
Add Optional to remaining types (#37808)
2025-04-28 14:20:45 +01: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
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
canine
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
chameleon
[vlm] adjust max length for special tokens (#37342)
2025-04-16 20:49:20 +02:00
chinese_clip
Add Fast Chinese-CLIP Processor (#37012)
2025-04-15 18:31:20 +02:00
clap
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
clip
🔴 Update CLIP vision attention to new attention interface (#37498)
2025-04-16 18:15:22 +02:00
clipseg
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
clvp
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
code_llama
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
codegen
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
cohere
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
cohere2
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
colpali
Fix typos (#37978)
2025-05-06 14:45:20 +01:00
conditional_detr
Add Fast Conditional-DETR Processor (#37071)
2025-04-15 18:33:34 +02:00
convbert
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
convnext
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
convnextv2
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02: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
enable csm integration cases on xpu, all passed (#38140)
2025-05-15 09:46:29 +02:00
ctrl
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
cvt
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
d_fine
[D-FINE] Update names (#37957)
2025-05-05 13:05:46 +01:00
dab_detr
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
dac
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
data2vec
Remove old code for PyTorch, Accelerator and tokenizers (#37234)
2025-04-10 20:54:21 +02:00
dbrx
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
deberta
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
deberta_v2
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
decision_transformer
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
deepseek_v3
Fix typos (#37978)
2025-05-06 14:45:20 +01:00
deformable_detr
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
deit
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
depth_anything
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
depth_pro
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
detr
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
diffllama
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
dinat
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
dinov2
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
dinov2_with_registers
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
distilbert
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
dit
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
donut
Add Fast Image Processor for Donut (#37081)
2025-04-14 16:24:01 +02:00
dpr
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
dpt
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
efficientnet
Add EfficientNet Image PreProcessor (#37055)
2025-04-16 21:59:24 +02:00
electra
Fix typos in strings and comments (#37784)
2025-04-25 13:47:25 +01:00
emu3
🔴 [VLM] Add base model without head (#37033)
2025-05-07 17:47:51 +02:00
encodec
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
encoder_decoder
🚨 rm already deprecated pad_to_max_length arg (#37617)
2025-05-01 15:21:55 +02:00
ernie
Remove old code for PyTorch, Accelerator and tokenizers (#37234)
2025-04-10 20:54:21 +02:00
esm
[ESM] Add flash-attention-2 backend for ESM-2 (#38023)
2025-05-16 14:11:56 +01:00
falcon
[test] update test_past_key_values_format (#37614)
2025-04-22 11:07:34 +01:00
falcon_mamba
enable several cases on XPU (#37516)
2025-04-16 11:01:04 +02:00
fastspeech2_conformer
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
flaubert
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
flava
Fix typos in strings and comments (#37799)
2025-04-28 11:39:11 +01:00
fnet
🚨 rm already deprecated pad_to_max_length arg (#37617)
2025-05-01 15:21:55 +02:00
focalnet
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
fsmt
Fix typos in strings and comments (#37910)
2025-05-01 14:58:58 +01:00
funnel
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
fuyu
🔴 [VLM] Add base model without head (#37033)
2025-05-07 17:47:51 +02:00
gemma
[test] update test_past_key_values_format (#37614)
2025-04-22 11:07:34 +01:00
gemma2
Fix typos in strings and comments (#37910)
2025-05-01 14:58:58 +01:00
gemma3
🔴 [VLM] Add base model without head (#37033)
2025-05-07 17:47:51 +02:00
git
🚨 🚨 Setup -> setupclass conversion (#37282)
2025-04-08 17:15:37 +01:00
glm
[test] update test_past_key_values_format (#37614)
2025-04-22 11:07:34 +01:00
glm4
Add glm4 (#37388)
2025-04-09 14:02:04 +02:00
glpn
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
got_ocr2
🔴 [VLM] Add base model without head (#37033)
2025-05-07 17:47:51 +02:00
gpt2
Add AMD expectation to test_gpt2_sample (#38079)
2025-05-12 16:51:21 +02:00
gpt_bigcode
Remove head mask in generative models (#35786)
2025-05-15 10:44:19 +02:00
gpt_neo
Fix typos in strings and comments (#37799)
2025-04-28 11:39:11 +01:00
gpt_neox
Remove old code for PyTorch, Accelerator and tokenizers (#37234)
2025-04-10 20:54:21 +02:00
gpt_neox_japanese
Remove old code for PyTorch, Accelerator and tokenizers (#37234)
2025-04-10 20:54:21 +02:00
gpt_sw3
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
gptj
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
granite
enable 6 granite cases on xpu (#37569)
2025-04-22 17:55:02 +02:00
granite_speech
enable finegrained_fp8 and granite_speech cases on XPU (#38036)
2025-05-14 08:58:40 +00:00
granitemoe
enable 6 granite cases on xpu (#37569)
2025-04-22 17:55:02 +02:00
granitemoehybrid
Add GraniteMoeHybrid support for 4.0 (#37658)
2025-05-06 06:47:43 +02:00
granitemoeshared
enable 6 granite cases on xpu (#37569)
2025-04-22 17:55:02 +02:00
grounding_dino
Fix typos in strings and comments (#37799)
2025-04-28 11:39:11 +01:00
groupvit
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
helium
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +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
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
hubert
Remove old code for PyTorch, Accelerator and tokenizers (#37234)
2025-04-10 20:54:21 +02:00
ibert
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
idefics
[tests] Stricter generate + compilation test -- no recompilations allowed (#37629)
2025-04-22 11:12:18 +01:00
idefics2
[VLMs] fix flash-attention tests (#37603)
2025-04-24 11:48:11 +02:00
idefics3
[VLMs] fix flash-attention tests (#37603)
2025-04-24 11:48:11 +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
Bart: new cache format (#35314)
2025-05-16 13:26:54 +02:00
instructblip
[VLMs] support attention backends (#37576)
2025-05-08 18:18:54 +02:00
instructblipvideo
🔴 Video processors as a separate class (#35206)
2025-05-12 11:55:51 +02:00
internvl
🔴 Video processors as a separate class (#35206)
2025-05-12 11:55:51 +02:00
jamba
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
janus
[tests] Stricter generate + compilation test -- no recompilations allowed (#37629)
2025-04-22 11:12:18 +01:00
jetmoe
[test] update test_past_key_values_format (#37614)
2025-04-22 11:07:34 +01:00
kosmos2
[VLMs] support attention backends (#37576)
2025-05-08 18:18:54 +02:00
layoutlm
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
layoutlmv2
🚨 rm already deprecated pad_to_max_length arg (#37617)
2025-05-01 15:21:55 +02:00
layoutlmv3
🚨 rm already deprecated pad_to_max_length arg (#37617)
2025-05-01 15:21:55 +02:00
layoutxlm
🚨 rm already deprecated pad_to_max_length arg (#37617)
2025-05-01 15:21:55 +02:00
led
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
levit
Add Fast LeViT Processor (#37154)
2025-04-14 17:07:36 +02:00
lilt
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
llama
Fix some GPU OOM after #37553 (#37591)
2025-04-18 10:09:19 +02:00
llama4
[processor] clean up mulitmodal tests (#37362)
2025-04-11 13:32:19 +02:00
llava
🔴 [VLM] Add base model without head (#37033)
2025-05-07 17:47:51 +02:00
llava_next
[bug] fix llava processor to calculate unpadding size correctly (#37988)
2025-05-13 13:49:09 +00:00
llava_next_video
[video processor] fix tests (#38104)
2025-05-14 10:24:07 +00:00
llava_onevision
[bug] fix llava processor to calculate unpadding size correctly (#37988)
2025-05-13 13:49:09 +00:00
longformer
Fix typos in strings and comments (#37799)
2025-04-28 11:39:11 +01:00
longt5
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
luke
Fix typos in strings and comments (#37799)
2025-04-28 11:39:11 +01:00
lxmert
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
m2m_100
Bart: new cache format (#35314)
2025-05-16 13:26:54 +02:00
mamba
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
mamba2
Add config validation and style tweaks (#37589)
2025-05-14 12:22:10 +00:00
marian
Bart: new cache format (#35314)
2025-05-16 13:26:54 +02:00
markuplm
🚨 rm already deprecated pad_to_max_length arg (#37617)
2025-05-01 15:21:55 +02:00
mask2former
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
maskformer
Fix typos in strings and comments (#37799)
2025-04-28 11:39:11 +01:00
mbart
Bart: new cache format (#35314)
2025-05-16 13:26:54 +02: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
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
mimi
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
mistral
[test] update test_past_key_values_format (#37614)
2025-04-22 11:07:34 +01:00
mistral3
🔴 Video processors as a separate class (#35206)
2025-05-12 11:55:51 +02:00
mixtral
[test] update test_past_key_values_format (#37614)
2025-04-22 11:07:34 +01:00
mlcd
Add MLCD model (#36182)
2025-04-15 11:33:09 +01:00
mllama
🔴 [VLM] Add base model without head (#37033)
2025-05-07 17:47:51 +02:00
mluke
Fix typos in strings and comments (#37799)
2025-04-28 11:39:11 +01:00
mobilebert
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
mobilenet_v1
Add Fast Image Processor for MobileNetV1 (#37111)
2025-04-23 15:55:41 -04:00
mobilenet_v2
Add Fast Mobilenet-V2 Processor (#37113)
2025-04-14 17:08:47 +02:00
mobilevit
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
mobilevitv2
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
modernbert
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
moonshine
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
moshi
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
mpnet
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
mpt
fix mpt test of different outputs from cuda (#37691)
2025-04-25 18:04:56 +02:00
mra
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
mt5
Fix mt5 test on AMD devices (#38081)
2025-05-12 16:59:00 +02:00
musicgen
Remove head mask in generative models (#35786)
2025-05-15 10:44:19 +02:00
musicgen_melody
Remove head mask in generative models (#35786)
2025-05-15 10:44:19 +02:00
mvp
Fix typos in strings and comments (#37799)
2025-04-28 11:39:11 +01:00
myt5
🚨 🚨 Setup -> setupclass conversion (#37282)
2025-04-08 17:15:37 +01:00
nemotron
[tests] fix test_nemotron_8b_generation_sdpa (#37665)
2025-04-24 11:28:35 +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
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
nystromformer
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
olmo
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
olmo2
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
olmoe
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
omdet_turbo
🚨 🚨 Setup -> setupclass conversion (#37282)
2025-04-08 17:15:37 +01:00
oneformer
Fix OneFormer integration test (#38016)
2025-05-12 16:02:41 +02:00
openai
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
opt
Remove head mask in generative models (#35786)
2025-05-15 10:44:19 +02:00
owlv2
🚨 🚨 Setup -> setupclass conversion (#37282)
2025-04-08 17:15:37 +01:00
owlvit
Add Fast owlvit Processor (#37164)
2025-04-14 17:58:09 +02:00
paligemma
🔴 [VLM] Add base model without head (#37033)
2025-05-07 17:47:51 +02:00
paligemma2
🔴 [VLM] Add base model without head (#37033)
2025-05-07 17:47:51 +02:00
patchtsmixer
Add Optional to remaining types (#37808)
2025-04-28 14:20:45 +01:00
patchtst
Force torch>=2.6 with torch.load to avoid vulnerability issue (#37785)
2025-04-25 16:57:09 +02:00
pegasus
Bart: new cache format (#35314)
2025-05-16 13:26:54 +02:00
pegasus_x
Bart: new cache format (#35314)
2025-05-16 13:26:54 +02:00
perceiver
Add Fast Image Processor for Perceiver (#37176)
2025-04-14 13:49:13 +02:00
persimmon
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
phi
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
phi3
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
phi4_multimodal
Refactor image processor phi4 (#36976)
2025-05-12 15:13:40 -04:00
phimoe
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
phobert
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
pix2struct
Fix typos in strings and comments (#37799)
2025-04-28 11:39:11 +01:00
pixtral
🔴 Video processors as a separate class (#35206)
2025-05-12 11:55:51 +02:00
plbart
Bart: new cache format (#35314)
2025-05-16 13:26:54 +02:00
poolformer
Add Fast Image Processor for PoolFormer (#37182)
2025-04-23 15:55:33 -04:00
pop2piano
Fix typos in strings and comments (#37799)
2025-04-28 11:39:11 +01:00
prompt_depth_anything
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
prophetnet
Fix typos in strings and comments (#37799)
2025-04-28 11:39:11 +01:00
pvt
Add Fast PVT Processor (#37204)
2025-04-23 15:55:20 -04:00
pvt_v2
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
qwen2
Fix Qwen models export with torch 2.7 (#37985)
2025-05-07 09:13:08 +02:00
qwen2_5_omni
[video processor] fix tests (#38104)
2025-05-14 10:24:07 +00:00
qwen2_5_vl
🔴 Video processors as a separate class (#35206)
2025-05-12 11:55:51 +02:00
qwen2_audio
[processor] clean up mulitmodal tests (#37362)
2025-04-11 13:32:19 +02:00
qwen2_moe
Fix typos in strings and comments (#37799)
2025-04-28 11:39:11 +01:00
qwen2_vl
Fix temporal padding in Qwen2VLImageProcessor when the number of frames is not divisible by temporal_patch_size (#38076)
2025-05-14 12:28:21 +02:00
qwen3
Fix Qwen models export with torch 2.7 (#37985)
2025-05-07 09:13:08 +02:00
qwen3_moe
Fix typos in strings and comments (#37799)
2025-04-28 11:39:11 +01:00
rag
Fix typos in strings and comments (#37799)
2025-04-28 11:39:11 +01:00
recurrent_gemma
Fixes #37219 : RecurrentGemma crashes for inputs longer than sliding window length (#37613)
2025-04-22 12:21:16 +02:00
reformer
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
regnet
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
rembert
Fix typos in strings and comments (#37784)
2025-04-25 13:47:25 +01:00
resnet
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
roberta
Fix typos in strings and comments (#37799)
2025-04-28 11:39:11 +01:00
roberta_prelayernorm
Fix typos in strings and comments (#37784)
2025-04-25 13:47:25 +01:00
roc_bert
Remove old code for PyTorch, Accelerator and tokenizers (#37234)
2025-04-10 20:54:21 +02:00
roformer
Remove old code for PyTorch, Accelerator and tokenizers (#37234)
2025-04-10 20:54:21 +02:00
rt_detr
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
rt_detr_v2
enable 6 rt_detr_v2 cases on xpu (#37548)
2025-04-16 11:23:56 +02:00
rwkv
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
sam
Fix typos in strings and comments (#37799)
2025-04-28 11:39:11 +01:00
sam_hq
Fix typos in strings and comments (#37910)
2025-05-01 14:58:58 +01:00
seamless_m4t
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
seamless_m4t_v2
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
segformer
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
seggpt
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
sew
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
sew_d
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
shieldgemma2
[chat-template] Unify tests and clean up 🧼 (#37275)
2025-04-10 14:42:32 +02:00
siglip
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
siglip2
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
smolvlm
[smolvlm] skip the test (#38099)
2025-05-13 12:50:43 +00:00
speech_encoder_decoder
enhance require_deterministic_for_xpu (#37437)
2025-04-11 08:06:08 +02:00
speech_to_text
[speech2text] fix init of sinusoidal embeddings (#37931)
2025-05-06 14:49:00 +01:00
speecht5
Flag SpeechT5 flaky test (#37587)
2025-04-18 11:35:46 +02:00
splinter
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
squeezebert
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
stablelm
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
starcoder2
[test] update test_past_key_values_format (#37614)
2025-04-22 11:07:34 +01:00
superglue
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
superpoint
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
swiftformer
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
swin
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
swin2sr
Add Swin2SR ImageProcessorFast (#37169)
2025-05-07 12:20:16 -04:00
swinv2
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
switch_transformers
Correctly drop tokens in SwitchTransformer (#37123)
2025-04-10 16:58:57 +02:00
t5
Fix typos in strings and comments (#37799)
2025-04-28 11:39:11 +01:00
table_transformer
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
tapas
🚨 rm already deprecated pad_to_max_length arg (#37617)
2025-05-01 15:21:55 +02:00
textnet
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
time_series_transformer
Force torch>=2.6 with torch.load to avoid vulnerability issue (#37785)
2025-04-25 16:57:09 +02:00
timesfm
[TimesFM] use the main revison instead of revision for integration test (#37558)
2025-04-17 11:26:03 +02:00
timesformer
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
timm_backbone
Small fix on context manager detection (#37562)
2025-04-17 15:39:44 +02:00
timm_wrapper
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
trocr
🚨 🚨 Setup -> setupclass conversion (#37282)
2025-04-08 17:15:37 +01:00
tvp
Add Optional to remaining types (#37808)
2025-04-28 14:20:45 +01:00
udop
🚨 rm already deprecated pad_to_max_length arg (#37617)
2025-05-01 15:21:55 +02:00
umt5
Fix typos in strings and comments (#37799)
2025-04-28 11:39:11 +01:00
unispeech
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
unispeech_sat
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
univnet
chore: fix typos in the tests directory (#36813)
2025-03-21 10:20:05 +01:00
upernet
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
video_llava
[video processor] fix tests (#38104)
2025-05-14 10:24:07 +00:00
videomae
Force torch>=2.6 with torch.load to avoid vulnerability issue (#37785)
2025-04-25 16:57:09 +02:00
vilt
Add Fast Image Processor for vilt (#37304)
2025-05-13 15:40:53 +00:00
vipllava
🔴 [VLM] Add base model without head (#37033)
2025-05-07 17:47:51 +02:00
vision_encoder_decoder
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
vision_text_dual_encoder
🚨 🚨 Setup -> setupclass conversion (#37282)
2025-04-08 17:15:37 +01:00
visual_bert
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
vit
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
vit_mae
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +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
Fast image processor for VitMatte added and bug in slow version fixed (#37616)
2025-04-28 14:51:50 -04:00
vitpose
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
vitpose_backbone
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
vits
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
vivit
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
wav2vec2
Remove old code for PyTorch, Accelerator and tokenizers (#37234)
2025-04-10 20:54:21 +02:00
wav2vec2_bert
🚨 🚨 Setup -> setupclass conversion (#37282)
2025-04-08 17:15:37 +01:00
wav2vec2_conformer
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
wav2vec2_phoneme
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
wav2vec2_with_lm
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
wavlm
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
whisper
Remove head mask in generative models (#35786)
2025-05-15 10:44:19 +02:00
x_clip
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
xglm
Fix typos in strings and comments (#37799)
2025-04-28 11:39:11 +01:00
xlm
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
xlm_roberta
Fix typos in strings and comments (#37799)
2025-04-28 11:39:11 +01:00
xlm_roberta_xl
Remove old code for PyTorch, Accelerator and tokenizers (#37234)
2025-04-10 20:54:21 +02:00
xlnet
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
xmod
Remove old code for PyTorch, Accelerator and tokenizers (#37234)
2025-04-10 20:54:21 +02:00
yolos
Add Fast Yolos Processor (#37292)
2025-04-15 14:23:08 +02:00
yoso
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
zamba
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
zamba2
Fix typos in strings and comments (#37799)
2025-04-28 11:39:11 +01:00
zoedepth
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
__init__.py
Move test model folders (#17034)
2022-05-03 14:42:02 +02:00