SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity
Files
211f2b08751cdee4bd6b84892b91f4a18dbfe305
HuggingFace_transformer/tests/models
History
Anton Vlasjuk d95c864a25 🔴🔴🔴 [Attention] Refactor Attention Interface for Bart-based Models (#38108) 
* starting attn refactor for encoder decoder models via bart (eager + sdpa)

* flash attention works, remove unnecessary code

* flex attention support for bart!, gotta check if the renaming is not too aggressive

* some comments

* skip flex grad test for standalone as done with the other test

* revert flex attn rename (for now), sdpa simplify, and todos

* more todos

* refactor mask creation for reuse

* modular attempt at biogpt

* first batch of other models

* fix attn dropout

* fix autoformer copies

* hubert

* another batch of models

* copies/style + last round of bart models --> whisper next?

* remove unnecessary _reshape function and remove copy to whisper

* add skip for decoder-only models out of enc-dec (same as in bart)

* bring back licences

* remove comment, added to pr read instead

* mostly docs

* disable sew flex attn as it's unclear attn mask for now

* oops

* test fixes for enc-dec

* torch fx fixes + try at flex attn

* skip on mbart

* some more fixes

* musicgen skip / delete old attn class logic + sdpa compose compile skip

* disable flex attn for musicgen, not worth the effort

* more fixes and style

* flex attention test for dropout and encoder decoder that dont have main input names

* informer fixes

* the weirdest thing I've encountered yet...

* style

* remove empty tensor attempt, found core root in previous commits

* disable time series due to tests being very text centric on inputs

* add speech to text to be ignoring the other attns, also due to tests

* update docs

* remaining issues resolved ?

* update docs for current state --> nllb moe and pegasus x sdpa is questionable :D

* some models have not set the is_causal flag...

* change dtype in softmax tol old behaviour + some modular fixes

* I hate it but it is what it is

* fixes from main for bart

* forgot this one

* some model fixes

* style

* current status

* marian works now

* fixing some copies

* some copy fixes + time series x informer

* last models possibly and fixes on style/copies

* some post merge fixes

* more fixes

* make attention interface callable and move warnings there

* style lol

* add comment to "unsupported"

* remove callable interface and change interface warnings + some copies

* fix

* ternary is ugly af, make it simpler

* how did that happen

* fix flex attn test

* failing the test

* no more fallback! fixing copies next

* style + attn fixed

* fixing copies and mask creation

* wrong copy

* fixup tests and disable flex attn for now

* fixup last tests?
2025-05-22 17:12:58 +02: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
Disable torchscript tests for AriaForConditionalGenerationModelTest (#38225)
2025-05-20 14:37:55 +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
[tests] remove test_sdpa_equivalence (redundant) (#37911)
2025-05-16 18:37:27 +01:00
bamba
Add padding-free to bamba (#35861)
2025-05-20 17:13:59 +02:00
bark
enable cpu offloading for Bark on xpu (#37599)
2025-04-23 11:37:15 +02:00
bart
🔴🔴🔴 [Attention] Refactor Attention Interface for Bart-based Models (#38108)
2025-05-22 17:12:58 +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
🔴🔴🔴 [Attention] Refactor Attention Interface for Bart-based Models (#38108)
2025-05-22 17:12:58 +02:00
blenderbot_small
🔴🔴🔴 [Attention] Refactor Attention Interface for Bart-based Models (#38108)
2025-05-22 17:12:58 +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
enable misc cases on XPU & use device agnostic APIs for cases in tests (#38192)
2025-05-20 10:09:01 +02:00
cohere2
[tests] remove test_sdpa_equivalence (redundant) (#37911)
2025-05-16 18:37:27 +01:00
colpali
enable misc cases on XPU & use device agnostic APIs for cases in tests (#38192)
2025-05-20 10:09:01 +02: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
🔴🔴🔴 [Attention] Refactor Attention Interface for Bart-based Models (#38108)
2025-05-22 17:12:58 +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
[tests] remove test_sdpa_equivalence (redundant) (#37911)
2025-05-16 18:37:27 +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
🔴🔴🔴 [Attention] Refactor Attention Interface for Bart-based Models (#38108)
2025-05-22 17:12:58 +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_h1
[Falcon H1] Fix Typo in Integration Test (#38256)
2025-05-21 11:25:26 +02: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
🚨🚨[core] Completely rewrite the masking logic for all attentions (#37866)
2025-05-22 11:38:26 +02:00
gemma2
🚨🚨[core] Completely rewrite the masking logic for all attentions (#37866)
2025-05-22 11:38:26 +02:00
gemma3
🚨🚨[core] Completely rewrite the masking logic for all attentions (#37866)
2025-05-22 11:38:26 +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
🔴🔴🔴 [Attention] Refactor Attention Interface for Bart-based Models (#38108)
2025-05-22 17:12:58 +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
enable misc cases on XPU & use device agnostic APIs for cases in tests (#38192)
2025-05-20 10:09:01 +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
Minor llama4 fixes (#38123)
2025-05-20 13:15:54 +00: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
fix multi-image case for llava-onevision (#38084)
2025-05-21 11:50:46 +02: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
🔴🔴🔴 [Attention] Refactor Attention Interface for Bart-based Models (#38108)
2025-05-22 17:12:58 +02:00
mamba
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
mamba2
Mamba2 remove unecessary test parameterization (#38227)
2025-05-20 13:54:04 +00:00
marian
🔴🔴🔴 [Attention] Refactor Attention Interface for Bart-based Models (#38108)
2025-05-22 17:12:58 +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
🔴🔴🔴 [Attention] Refactor Attention Interface for Bart-based Models (#38108)
2025-05-22 17:12:58 +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
🔴🔴🔴 [Attention] Refactor Attention Interface for Bart-based Models (#38108)
2025-05-22 17:12:58 +02:00
musicgen_melody
🔴🔴🔴 [Attention] Refactor Attention Interface for Bart-based Models (#38108)
2025-05-22 17:12:58 +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] remove test_sdpa_equivalence (redundant) (#37911)
2025-05-16 18:37:27 +01: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
Make HF implementation match original OLMo 2 models for lower precisions (#38131)
2025-05-19 15:35:23 +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
🚨🚨[core] Completely rewrite the masking logic for all attentions (#37866)
2025-05-22 11:38:26 +02: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
🔴🔴🔴 [Attention] Refactor Attention Interface for Bart-based Models (#38108)
2025-05-22 17:12:58 +02:00
pegasus_x
🔴🔴🔴 [Attention] Refactor Attention Interface for Bart-based Models (#38108)
2025-05-22 17:12:58 +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
enable misc cases on XPU & use device agnostic APIs for cases in tests (#38192)
2025-05-20 10:09:01 +02: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
Add args support for fast image processors (#37018)
2025-05-16 12:01:46 -04:00
plbart
🔴🔴🔴 [Attention] Refactor Attention Interface for Bart-based Models (#38108)
2025-05-22 17:12:58 +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
[compile] re-enable for Qwen-VL models (#38127)
2025-05-21 09:50:39 +00: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
[compile] re-enable for Qwen-VL models (#38127)
2025-05-21 09:50:39 +00: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
[SAM-HQ] Update names in the docs (#38058)
2025-05-19 09:21:14 -07: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
🔴🔴🔴 [Attention] Refactor Attention Interface for Bart-based Models (#38108)
2025-05-22 17:12:58 +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
🔴🔴🔴 [Attention] Refactor Attention Interface for Bart-based Models (#38108)
2025-05-22 17:12:58 +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
🔴🔴🔴 [Attention] Refactor Attention Interface for Bart-based Models (#38108)
2025-05-22 17:12:58 +02:00
unispeech_sat
🔴🔴🔴 [Attention] Refactor Attention Interface for Bart-based Models (#38108)
2025-05-22 17:12:58 +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
🔴🔴🔴 [Attention] Refactor Attention Interface for Bart-based Models (#38108)
2025-05-22 17:12:58 +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
Add args support for fast image processors (#37018)
2025-05-16 12:01:46 -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
🔴🔴🔴 [Attention] Refactor Attention Interface for Bart-based Models (#38108)
2025-05-22 17:12:58 +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
[whisper] move processor test into processor test file 🧹 (#38266)
2025-05-22 10:07:11 +01: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