transformers/tests/models
Yih-Dar c87058beb8
Fix llava_onevision tests (#38791)
* fix

* fix

* fix

* fix

---------

Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
2025-06-12 15:06:49 +02:00
..
albert Deprecate TF + JAX (#38758) 2025-06-11 17:28:06 +01: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 Don't run AriaForConditionalGenerationModelTest on CircleCI (#38615) 2025-06-06 11:30:31 +02:00
audio_spectrogram_transformer Use Python 3.9 syntax in tests (#37343) 2025-04-08 14:12:08 +02:00
auto Deprecate TF + JAX (#38758) 2025-06-11 17:28:06 +01: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 Fix aya_vision test (#38674) 2025-06-09 22:18:52 +02:00
bamba switch to device agnostic device calling for test cases (#38247) 2025-05-26 10:18:53 +02:00
bark enable more test cases on xpu (#38572) 2025-06-06 09:29:51 +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 Deprecate TF + JAX (#38758) 2025-06-11 17:28:06 +01:00
bert Deprecate TF + JAX (#38758) 2025-06-11 17:28:06 +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 Deprecate TF + JAX (#38758) 2025-06-11 17:28:06 +01:00
bigbird_pegasus [generation] bring back tests on vision models (#38603) 2025-06-06 08:23:15 +00: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 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 Deprecate TF + JAX (#38758) 2025-06-11 17:28:06 +01:00
blip_2 [generation] bring back tests on vision models (#38603) 2025-06-06 08:23:15 +00:00
bloom Deprecate TF + JAX (#38758) 2025-06-11 17:28:06 +01: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 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 Update some tests for torch 2.7.1 (#38701) 2025-06-10 11:46:52 +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 Deprecate TF + JAX (#38758) 2025-06-11 17:28:06 +01: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 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 enable more test cases on xpu (#38572) 2025-06-06 09:29:51 +02:00
cohere2 Fix Gemma2IntegrationTest (#38492) 2025-06-02 22:45:09 +02:00
colpali Add ColQwen2 to 🤗 transformers (#35778) 2025-06-02 12:58:01 +00:00
colqwen2 Update some tests for torch 2.7.1 (#38701) 2025-06-10 11:46:52 +02:00
conditional_detr 🚨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
convbert Deprecate TF + JAX (#38758) 2025-06-11 17:28:06 +01:00
convnext Deprecate TF + JAX (#38758) 2025-06-11 17:28:06 +01: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 Update CsmForConditionalGenerationIntegrationTest (#38424) 2025-05-28 10:20:43 +02:00
ctrl Deprecate TF + JAX (#38758) 2025-06-11 17:28:06 +01:00
cvt Deprecate TF + JAX (#38758) 2025-06-11 17:28:06 +01:00
d_fine 🚨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
dab_detr 🚨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
dac Use Python 3.9 syntax in tests (#37343) 2025-04-08 14:12:08 +02:00
data2vec Deprecate TF + JAX (#38758) 2025-06-11 17:28:06 +01:00
dbrx 🚨 🚨 Inherited CausalLM Tests (#37590) 2025-05-23 18:29:31 +01: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_v3 [FlexAttn] Fix models with unique characteristics (#38433) 2025-06-04 13:37:28 +02:00
deformable_detr enable more test cases on xpu (#38572) 2025-06-06 09:29:51 +02:00
deit Deprecate TF + JAX (#38758) 2025-06-11 17:28:06 +01:00
depth_anything Skip some export tests on torch 2.7 (#38677) 2025-06-12 12:47:15 +02:00
depth_pro Use Python 3.9 syntax in tests (#37343) 2025-04-08 14:12:08 +02:00
detr enable more test cases on xpu (#38572) 2025-06-06 09:29:51 +02:00
diffllama Remove deprecated use_flash_attention_2 parameter (#37131) 2025-06-02 11:06:25 +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 Use Python 3.9 syntax in tests (#37343) 2025-04-08 14:12:08 +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
donut 🚨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
dpr Deprecate TF + JAX (#38758) 2025-06-11 17:28:06 +01:00
dpt Skip some export tests on torch 2.7 (#38677) 2025-06-12 12:47:15 +02: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 update emu3 test (#38543) 2025-06-03 11:02:01 +02:00
encodec Use Python 3.9 syntax in tests (#37343) 2025-04-08 14:12:08 +02:00
encoder_decoder Deprecate TF + JAX (#38758) 2025-06-11 17:28:06 +01:00
ernie Remove old code for PyTorch, Accelerator and tokenizers (#37234) 2025-04-10 20:54:21 +02:00
esm Deprecate TF + JAX (#38758) 2025-06-11 17:28:06 +01:00
falcon 🚨 🚨 Inherited CausalLM Tests (#37590) 2025-05-23 18:29:31 +01:00
falcon_h1 [Falcon H1] Fix slow path forward pass (#38320) 2025-05-26 15:30:35 +02:00
falcon_mamba enable several cases on XPU (#37516) 2025-04-16 11:01:04 +02:00
fastspeech2_conformer 🚨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
flaubert Deprecate TF + JAX (#38758) 2025-06-11 17:28:06 +01:00
flava 🚨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
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 Deprecate TF + JAX (#38758) 2025-06-11 17:28:06 +01:00
fuyu 🔴 [VLM] Add base model without head (#37033) 2025-05-07 17:47:51 +02:00
gemma Deprecate TF + JAX (#38758) 2025-06-11 17:28:06 +01:00
gemma2 enable more test cases on xpu (#38572) 2025-06-06 09:29:51 +02:00
gemma3 enable more test cases on xpu (#38572) 2025-06-06 09:29:51 +02:00
git 🚨 🚨 Setup -> setupclass conversion (#37282) 2025-04-08 17:15:37 +01:00
glm switch to device agnostic device calling for test cases (#38247) 2025-05-26 10:18:53 +02:00
glm4 enable more test cases on xpu (#38572) 2025-06-06 09:29:51 +02:00
glpn 🚨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
got_ocr2 [tests] expand flex-attn test for vision models (#38434) 2025-06-03 07:40:44 +00:00
gpt_bigcode Remove head mask in generative models (#35786) 2025-05-15 10:44:19 +02:00
gpt_neo Deprecate TF + JAX (#38758) 2025-06-11 17:28:06 +01:00
gpt_neox 🚨 🚨 Inherited CausalLM Tests (#37590) 2025-05-23 18:29:31 +01: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 Deprecate TF + JAX (#38758) 2025-06-11 17:28:06 +01: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 enable finegrained_fp8 and granite_speech cases on XPU (#38036) 2025-05-14 08:58:40 +00: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 enable more test cases on xpu (#38572) 2025-06-06 09:29:51 +02:00
groupvit Deprecate TF + JAX (#38758) 2025-06-11 17:28:06 +01:00
helium switch to device agnostic device calling for test cases (#38247) 2025-05-26 10:18:53 +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 🚨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
hubert Deprecate TF + JAX (#38758) 2025-06-11 17:28:06 +01:00
ibert Use Python 3.9 syntax in tests (#37343) 2025-04-08 14:12:08 +02:00
idefics Deprecate TF + JAX (#38758) 2025-06-11 17:28:06 +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 🚨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 [VLMs] support attention backends (#37576) 2025-05-08 18:18:54 +02:00
instructblipvideo [video processors] support frame sampling within processors (#38105) 2025-06-12 09:34:30 +00:00
internvl [video processors] support frame sampling within processors (#38105) 2025-06-12 09:34:30 +00:00
jamba switch to device agnostic device calling for test cases (#38247) 2025-05-26 10:18:53 +02:00
janus [janus] Fix failing tests on mi3XX (#38426) 2025-06-04 09:38:10 +02:00
jetmoe 🚨 🚨 Inherited CausalLM Tests (#37590) 2025-05-23 18:29:31 +01:00
kosmos2 [VLMs] support attention backends (#37576) 2025-05-08 18:18:54 +02:00
layoutlm Deprecate TF + JAX (#38758) 2025-06-11 17:28:06 +01:00
layoutlmv2 🚨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
layoutlmv3 Deprecate TF + JAX (#38758) 2025-06-11 17:28:06 +01:00
layoutxlm 🚨 rm already deprecated pad_to_max_length arg (#37617) 2025-05-01 15:21:55 +02:00
led Deprecate TF + JAX (#38758) 2025-06-11 17:28:06 +01: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 Deprecate TF + JAX (#38758) 2025-06-11 17:28:06 +01:00
llama4 enable more test cases on xpu (#38572) 2025-06-06 09:29:51 +02:00
llava [llava] fix integration tests with Siglip (#38732) 2025-06-11 08:09:16 +00: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 llava_onevision tests (#38791) 2025-06-12 15:06:49 +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 🚨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
lxmert Deprecate TF + JAX (#38758) 2025-06-11 17:28:06 +01: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 Deprecate TF + JAX (#38758) 2025-06-11 17:28:06 +01: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 🚨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
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 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
minimax Fix MiniMax (docs and integration tests checkpoint) (#38575) 2025-06-06 08:43:11 +02:00
mistral Deprecate TF + JAX (#38758) 2025-06-11 17:28:06 +01:00
mistral3 🔴 Video processors as a separate class (#35206) 2025-05-12 11:55:51 +02:00
mixtral switch to device agnostic device calling for test cases (#38247) 2025-05-26 10:18:53 +02:00
mlcd Add MLCD model (#36182) 2025-04-15 11:33:09 +01:00
mllama fix spelling errors (#38608) 2025-06-05 13:57:23 +01:00
mluke Fix typos in strings and comments (#37799) 2025-04-28 11:39:11 +01:00
mobilebert Deprecate TF + JAX (#38758) 2025-06-11 17:28:06 +01: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 Deprecate TF + JAX (#38758) 2025-06-11 17:28:06 +01: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 Skip torchscript tests for 2 models (#38643) 2025-06-06 20:17:37 +02:00
moshi Use Python 3.9 syntax in tests (#37343) 2025-04-08 14:12:08 +02:00
mpnet Deprecate TF + JAX (#38758) 2025-06-11 17:28:06 +01: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 Deprecate TF + JAX (#38758) 2025-06-11 17:28:06 +01:00
musicgen [tests] expand flex-attn test for vision models (#38434) 2025-06-03 07:40:44 +00:00
musicgen_melody [tests] expand flex-attn test for vision models (#38434) 2025-06-03 07:40:44 +00: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 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 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 enable more test cases on xpu (#38572) 2025-06-06 09:29:51 +02:00
oneformer Fix OneFormer integration test (#38016) 2025-05-12 16:02:41 +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 🚨 🚨 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 [paligemma] fix processor with suffix (#38365) 2025-05-27 11:31:56 +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 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 🚨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
persimmon 🚨 🚨 Inherited CausalLM Tests (#37590) 2025-05-23 18:29:31 +01:00
phi 🚨 🚨 Inherited CausalLM Tests (#37590) 2025-05-23 18:29:31 +01:00
phi3 🚨 🚨 Inherited CausalLM Tests (#37590) 2025-05-23 18:29:31 +01:00
phi4_multimodal Fix incorrect batching audio index calculation for Phi-4-Multimodal (#38103) 2025-05-26 12:41:31 +00: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 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 Skip some export tests on torch 2.7 (#38677) 2025-06-12 12:47:15 +02:00
prophetnet [generation] bring back tests on vision models (#38603) 2025-06-06 08:23:15 +00: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 🚨 🚨 Inherited CausalLM Tests (#37590) 2025-05-23 18:29:31 +01:00
qwen2_5_omni Fix qwen_2_5 omni (#38658) 2025-06-12 14:43:54 +02:00
qwen2_5_vl [video processors] support frame sampling within processors (#38105) 2025-06-12 09:34:30 +00:00
qwen2_audio [Tests] Clean up test cases for few models (#38315) 2025-05-29 08:21:28 +00:00
qwen2_moe Fix MoE gradient test (#38438) 2025-05-28 16:44:20 +01:00
qwen2_vl [video processors] support frame sampling within processors (#38105) 2025-06-12 09:34:30 +00:00
qwen3 Update some tests for torch 2.7.1 (#38701) 2025-06-10 11:46:52 +02:00
qwen3_moe Fix MoE gradient test (#38438) 2025-05-28 16:44:20 +01:00
rag Deprecate TF + JAX (#38758) 2025-06-11 17:28:06 +01:00
recurrent_gemma 🚨 🚨 Inherited CausalLM Tests (#37590) 2025-05-23 18:29:31 +01:00
reformer Use Python 3.9 syntax in tests (#37343) 2025-04-08 14:12:08 +02:00
regnet Deprecate TF + JAX (#38758) 2025-06-11 17:28:06 +01:00
rembert Deprecate TF + JAX (#38758) 2025-06-11 17:28:06 +01:00
resnet Deprecate TF + JAX (#38758) 2025-06-11 17:28:06 +01: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 enable more test cases on xpu (#38572) 2025-06-06 09:29:51 +02:00
rt_detr_v2 🚨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
rwkv 🚨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
sam Deprecate TF + JAX (#38758) 2025-06-11 17:28:06 +01:00
sam_hq 🚨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
seamless_m4t [seamless_m4t] Skip some tests when speech is not available (#38430) 2025-06-02 09:17:28 +00:00
seamless_m4t_v2 [seamless_m4t] Skip some tests when speech is not available (#38430) 2025-06-02 09:17:28 +00:00
segformer Deprecate TF + JAX (#38758) 2025-06-11 17:28:06 +01: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 enable more test cases on xpu (#38572) 2025-06-06 09:29:51 +02:00
siglip [tests] expand flex-attn test for vision models (#38434) 2025-06-03 07:40:44 +00:00
siglip2 [tests] expand flex-attn test for vision models (#38434) 2025-06-03 07:40:44 +00:00
smolvlm [video processors] support frame sampling within processors (#38105) 2025-06-12 09:34:30 +00:00
speech_encoder_decoder Deprecate TF + JAX (#38758) 2025-06-11 17:28:06 +01:00
speech_to_text Deprecate TF + JAX (#38758) 2025-06-11 17:28:06 +01:00
speecht5 [generation] bring back tests on vision models (#38603) 2025-06-06 08:23:15 +00: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 🚨 🚨 Inherited CausalLM Tests (#37590) 2025-05-23 18:29:31 +01:00
starcoder2 🚨 🚨 Inherited CausalLM Tests (#37590) 2025-05-23 18:29:31 +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 Deprecate TF + JAX (#38758) 2025-06-11 17:28:06 +01:00
swin Deprecate TF + JAX (#38758) 2025-06-11 17:28:06 +01:00
swin2sr 🚨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
swinv2 🚨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
switch_transformers [generation] bring back tests on vision models (#38603) 2025-06-06 08:23:15 +00:00
t5 Deprecate TF + JAX (#38758) 2025-06-11 17:28:06 +01: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 Deprecate TF + JAX (#38758) 2025-06-11 17:28:06 +01: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 [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 [generation] bring back tests on vision models (#38603) 2025-06-06 08:23:15 +00:00
umt5 [generation] bring back tests on vision models (#38603) 2025-06-06 08:23:15 +00: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 Skip some export tests on torch 2.7 (#38677) 2025-06-12 12:47:15 +02:00
video_llava fix spelling errors (#38608) 2025-06-05 13:57:23 +01:00
videomae [tests] expand flex-attn test for vision models (#38434) 2025-06-03 07:40:44 +00:00
vilt 🚨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
vipllava [tests] expand flex-attn test for vision models (#38434) 2025-06-03 07:40:44 +00:00
vision_encoder_decoder Deprecate TF + JAX (#38758) 2025-06-11 17:28:06 +01:00
vision_text_dual_encoder Deprecate TF + JAX (#38758) 2025-06-11 17:28:06 +01:00
visual_bert 🚨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
vit Deprecate TF + JAX (#38758) 2025-06-11 17:28:06 +01:00
vit_mae Deprecate TF + JAX (#38758) 2025-06-11 17:28:06 +01: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 Skip some export tests on torch 2.7 (#38677) 2025-06-12 12:47:15 +02:00
vitpose Skip some export tests on torch 2.7 (#38677) 2025-06-12 12:47:15 +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 🚨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
vjepa2 Add V-JEPA 2 (#38746) 2025-06-11 15:00:08 +01:00
wav2vec2 Deprecate TF + JAX (#38758) 2025-06-11 17:28:06 +01: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 Deprecate TF + JAX (#38758) 2025-06-11 17:28:06 +01:00
x_clip 🚨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
xglm Deprecate TF + JAX (#38758) 2025-06-11 17:28:06 +01: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
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 🚨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
zamba2 [FlexAttn] Fix models with unique characteristics (#38433) 2025-06-04 13:37:28 +02:00
zoedepth Skip some export tests on torch 2.7 (#38677) 2025-06-12 12:47:15 +02:00
__init__.py Move test model folders (#17034) 2022-05-03 14:42:02 +02:00