transformers/tests/models
Vladislav Bronzov 4abeb50f6e
Add D-FINE Model into Transformers (#36261)
* copy the last changes from broken PR

* small format

* some fixes and refactoring after review

* format

* add config attr for loss

* some fixes and refactoring

* fix copies

* fix style

* add test for d-fine resnet

* fix decoder layer prop

* fix dummies

* format init

* remove extra print

* refactor modeling, move resnet into separate folder

* fix resnet config

* change resnet on hgnet_v2, add clamp into decoder

* fix init

* fix config doc

* fix init

* fix dummies

* fix config docs

* fix hgnet_v2 config typo

* format modular

* add image classification for hgnet, some refactoring

* format tests

* fix dummies

* fix init

* fix style

* fix init for hgnet v2

* fix index.md, add init rnage for hgnet

* fix conversion

* add missing attr to encoder

* add loss for d-fine, add additional output for rt-detr decoder

* tests and docs fixes

* fix rt_detr v2 conversion

* some fixes for loos and decoder output

* some fixes for loss

* small fix for converted modeling

* add n model config, some todo comments for modular

* convert script adjustments and fixes, small refact

* remove extra output for rt_detr

* make some outputs optionsl, fix conversion

* some posr merge fixes

* small fix

* last field fix

* fix not split for hgnet_v2

* disable parallelism test for hgnet_v2 image classification

* skip multi gpu for d-fine

* adjust after merge init

* remove extra comment

* fix repo name references

* small fixes for tests

* Fix checkpoint path

* Fix consistency

* Fixing docs

---------

Co-authored-by: Pavel Iakubovskii <qubvel@gmail.com>
2025-04-29 12:17:55 +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 Skip all AriaForConditionalGenerationIntegrationTest on T4 (#37746) 2025-04-24 14:11:56 +02:00
audio_spectrogram_transformer Use Python 3.9 syntax in tests (#37343) 2025-04-08 14:12:08 +02:00
auto 🚨 🚨 Allow saving and loading multiple "raw" chat template files (#36588) 2025-04-11 16:37:23 +01:00
autoformer Force torch>=2.6 with torch.load to avoid vulnerability issue (#37785) 2025-04-25 16:57:09 +02:00
aya_vision [VLMs] fix flash-attention tests (#37603) 2025-04-24 11:48:11 +02:00
bamba enable 6 modeling cases on XPU (#37571) 2025-04-18 12:28:08 +02:00
bark enable cpu offloading for Bark on xpu (#37599) 2025-04-23 11:37:15 +02:00
bart Use Python 3.9 syntax in tests (#37343) 2025-04-08 14:12:08 +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 Use Python 3.9 syntax in tests (#37343) 2025-04-08 14:12:08 +02: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 Use Python 3.9 syntax in tests (#37343) 2025-04-08 14:12:08 +02:00
biogpt Use Python 3.9 syntax in tests (#37343) 2025-04-08 14:12:08 +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 Use Python 3.9 syntax in tests (#37343) 2025-04-08 14:12:08 +02:00
blenderbot_small Use Python 3.9 syntax in tests (#37343) 2025-04-08 14:12:08 +02:00
blip 🚨 🚨 Setup -> setupclass conversion (#37282) 2025-04-08 17:15:37 +01:00
blip_2 enable blip2 and emu3 cases on XPU (#37662) 2025-04-22 18:37:09 +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 Simplify soft dependencies and update the dummy-creation process (#36827) 2025-04-11 11:08:36 +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
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 Add D-FINE Model into Transformers (#36261) 2025-04-29 12:17:55 +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 [test] update test_past_key_values_format (#37614) 2025-04-22 11:07:34 +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 enable blip2 and emu3 cases on XPU (#37662) 2025-04-22 18:37:09 +02:00
encodec Use Python 3.9 syntax in tests (#37343) 2025-04-08 14:12:08 +02:00
encoder_decoder fix 2 encoder_decoder issues on XPU (#37572) 2025-04-18 17:49:24 +02:00
ernie Remove old code for PyTorch, Accelerator and tokenizers (#37234) 2025-04-10 20:54:21 +02:00
esm Fix typos in strings and comments (#37784) 2025-04-25 13:47:25 +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 Use Python 3.9 syntax in tests (#37343) 2025-04-08 14:12:08 +02:00
focalnet Use Python 3.9 syntax in tests (#37343) 2025-04-08 14:12:08 +02:00
fsmt Use Python 3.9 syntax in tests (#37343) 2025-04-08 14:12:08 +02:00
funnel Use Python 3.9 syntax in tests (#37343) 2025-04-08 14:12:08 +02:00
fuyu [processor] clean up mulitmodal tests (#37362) 2025-04-11 13:32:19 +02:00
gemma [test] update test_past_key_values_format (#37614) 2025-04-22 11:07:34 +01:00
gemma2 Gemma3 is Torch Exportable (#37728) 2025-04-28 09:36:46 +02:00
gemma3 Fix typos in strings and comments (#37799) 2025-04-28 11:39:11 +01: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 [VLMs] fix flash-attention tests (#37603) 2025-04-24 11:48:11 +02:00
gpt_bigcode Use Python 3.9 syntax in tests (#37343) 2025-04-08 14:12:08 +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
gpt2 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 Add Granite Speech Support (#36801) 2025-04-11 18:52:00 +02:00
granitemoe enable 6 granite cases on xpu (#37569) 2025-04-22 17:55:02 +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 Force torch>=2.6 with torch.load to avoid vulnerability issue (#37785) 2025-04-25 16:57:09 +02:00
instructblip Fix typos in strings and comments (#37799) 2025-04-28 11:39:11 +01:00
instructblipvideo 🚨 🚨 Setup -> setupclass conversion (#37282) 2025-04-08 17:15:37 +01:00
internvl [internvl] fix chat template (#37656) 2025-04-23 16:56:36 +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 make test_snowman_image_captioning pass on XPU, by sharing same atol w/ ROCM (#37480) 2025-04-14 11:39:45 +02:00
layoutlm Use Python 3.9 syntax in tests (#37343) 2025-04-08 14:12:08 +02:00
layoutlmv2 Add Fast Image Processor for LayoutLMv2 (#37203) 2025-04-14 15:06:41 +02:00
layoutlmv3 Add Fast Image Processor for LayoutLMv3 (#37201) 2025-04-14 15:42:11 +02:00
layoutxlm 🚨 🚨 Setup -> setupclass conversion (#37282) 2025-04-08 17:15:37 +01: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 Fix typos in strings and comments (#37799) 2025-04-28 11:39:11 +01:00
llava_next Force torch>=2.6 with torch.load to avoid vulnerability issue (#37785) 2025-04-25 16:57:09 +02:00
llava_next_video [tests] Stricter generate + compilation test -- no recompilations allowed (#37629) 2025-04-22 11:12:18 +01:00
llava_onevision [VLMs] fix flash-attention tests (#37603) 2025-04-24 11:48:11 +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 Use Python 3.9 syntax in tests (#37343) 2025-04-08 14:12:08 +02:00
mamba Use Python 3.9 syntax in tests (#37343) 2025-04-08 14:12:08 +02:00
mamba2 Fix Mamba2 Grouped SSD Support in the torch_forward Path (#37533) 2025-04-16 22:16:01 +02:00
marian Use Python 3.9 syntax in tests (#37343) 2025-04-08 14:12:08 +02:00
markuplm Use Python 3.9 syntax in tests (#37343) 2025-04-08 14:12:08 +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 Use Python 3.9 syntax in tests (#37343) 2025-04-08 14:12:08 +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 [vlm] adjust max length for special tokens (#37342) 2025-04-16 20:49:20 +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 Fix typos in strings and comments (#37799) 2025-04-28 11:39:11 +01: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 typos in strings and comments (#37799) 2025-04-28 11:39:11 +01:00
musicgen Allow rocm systems to run these tests (#37278) 2025-04-10 13:33:01 +02:00
musicgen_melody Allow rocm systems to run these tests (#37278) 2025-04-10 13:33:01 +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 Use Python 3.9 syntax in tests (#37343) 2025-04-08 14:12:08 +02:00
openai Use Python 3.9 syntax in tests (#37343) 2025-04-08 14:12:08 +02:00
opt Force torch>=2.6 with torch.load to avoid vulnerability issue (#37785) 2025-04-25 16:57:09 +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 Fix typos in strings and comments (#37799) 2025-04-28 11:39:11 +01:00
paligemma2 Fix typos in strings and comments (#37799) 2025-04-28 11:39:11 +01: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 Use Python 3.9 syntax in tests (#37343) 2025-04-08 14:12:08 +02:00
pegasus_x Use Python 3.9 syntax in tests (#37343) 2025-04-08 14:12:08 +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 Detect and fix most _init_weights() issues - make it work for composite models (#37070) 2025-04-14 16:19:04 +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 Fix typos in strings and comments (#37799) 2025-04-28 11:39:11 +01:00
plbart Use Python 3.9 syntax in tests (#37343) 2025-04-08 14:12:08 +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 [test] update test_past_key_values_format (#37614) 2025-04-22 11:07:34 +01:00
qwen2_5_omni Qwen 2.5 Omni: apply video defaults (#37660) 2025-04-23 17:08:11 +02:00
qwen2_5_vl Fix qwen2_5 get_rope_index tensor device locations (#37597) 2025-04-24 16:04:38 +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 typos in strings and comments (#37799) 2025-04-28 11:39:11 +01:00
qwen3 [test] update test_past_key_values_format (#37614) 2025-04-22 11:07:34 +01: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 Samhq model addition (#35147) 2025-04-28 19:07:09 +02: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 Fix typos in strings and comments (#37799) 2025-04-28 11:39:11 +01:00
speech_encoder_decoder enhance require_deterministic_for_xpu (#37437) 2025-04-11 08:06:08 +02:00
speech_to_text 🚨 🚨 Setup -> setupclass conversion (#37282) 2025-04-08 17:15:37 +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 Use Python 3.9 syntax in tests (#37343) 2025-04-08 14:12:08 +02: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 Use Python 3.9 syntax in tests (#37343) 2025-04-08 14:12:08 +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 Fix typos in strings and comments (#37799) 2025-04-28 11:39:11 +01: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 [VLMs] fix flash-attention tests (#37603) 2025-04-24 11:48:11 +02:00
videomae Force torch>=2.6 with torch.load to avoid vulnerability issue (#37785) 2025-04-25 16:57:09 +02:00
vilt Use Python 3.9 syntax in tests (#37343) 2025-04-08 14:12:08 +02:00
vipllava [VLMs] fix flash-attention tests (#37603) 2025-04-24 11:48:11 +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 Fix typos in strings and comments (#37799) 2025-04-28 11:39: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