transformers/tests/models
Arthur c23a1c1932
Add-helium (#35669)
* Add the helium model.

* Add a missing helium.

* And add another missing helium.

* Use float for the rmsnorm mul.

* Add the Helium tokenizer converter.

* Add the pad token as suggested by Arthur.

* Update the RMSNorm + some other tweaks.

* Fix more rebase issues.

* fix copies and style

* fixes and add helium.md

* add missing tests

* udpate the backlink

* oups

* style

* update init, and expected results

* small fixes

* match test outputs

* style fixup, fix doc builder

* add dummies and we should be good to go!z

* update sdpa and fa2 documentation

---------

Co-authored-by: laurent <laurent.mazare@gmail.com>
2025-01-13 18:41:15 +01:00
..
albert Remove @slow for test_eager_matches_sdpa_inference (#34558) 2024-11-05 16:10:42 +01:00
align Load sub-configs from composite configs (#34410) 2024-11-05 11:34:01 +01:00
altclip Load sub-configs from composite configs (#34410) 2024-11-05 11:34:01 +01:00
aria Aurevoir PyTorch 1 (#35358) 2024-12-20 14:36:31 +01:00
audio_spectrogram_transformer [tests] fix "Tester object has no attribute '_testMethodName'" (#34910) 2024-12-13 14:33:45 +01:00
auto Fall back to slow image processor in ImageProcessingAuto when no fast processor available (#34785) 2024-12-15 14:00:36 -05:00
autoformer
bamba Add the Bamba Model (#34982) 2024-12-18 20:18:17 +01:00
bark
bart Tests: move generate tests to the right mixin and delete redundant tests (#34464) 2024-10-30 10:59:08 +00:00
barthez
bartpho
beit Add sdpa for Beit (#34941) 2024-12-17 14:44:47 +01:00
bert Tests: move generate tests to the right mixin and delete redundant tests (#34464) 2024-10-30 10:59:08 +00:00
bert_generation
bert_japanese
bertweet
big_bird [pytes collection] Fix flax test collection (#34004) 2024-10-07 18:11:13 +02:00
bigbird_pegasus Make pipeline able to load processor (#32514) 2024-10-09 16:46:11 +01:00
biogpt Generate: unify LogitsWarper and LogitsProcessor (#32626) 2024-08-16 11:20:41 +01:00
bit
blenderbot Forbid PretrainedConfig from saving generate parameters; Update deprecations in generate-related code 🧹 (#32659) 2024-08-23 11:12:53 +01:00
blenderbot_small Make pipeline able to load processor (#32514) 2024-10-09 16:46:11 +01:00
blip Load sub-configs from composite configs (#34410) 2024-11-05 11:34:01 +01:00
blip_2 [tests] make cuda-only tests device-agnostic (#35607) 2025-01-13 14:48:39 +01:00
bloom VLMs: enable generation tests (#33533) 2024-09-19 12:04:24 +02:00
bridgetower rename all test_processing_*.py to test_processor_*.py (#33878) 2024-10-02 16:43:43 +02:00
bros Make pipeline able to load processor (#32514) 2024-10-09 16:46:11 +01:00
byt5
camembert VLM: special multimodal Tokenizer (#34461) 2024-11-04 16:37:51 +01:00
canine
chameleon [Compile] Only test compiling model forward pass (#35658) 2025-01-13 13:43:29 +01:00
chinese_clip adding positional encoder changes and tests (#32600) 2024-09-25 19:05:01 +01:00
clap [tests] fix "Tester object has no attribute '_testMethodName'" (#34910) 2024-12-13 14:33:45 +01:00
clip [tests] fix "Tester object has no attribute '_testMethodName'" (#34910) 2024-12-13 14:33:45 +01:00
clipseg Load sub-configs from composite configs (#34410) 2024-11-05 11:34:01 +01:00
clvp [tests] fix "Tester object has no attribute '_testMethodName'" (#34910) 2024-12-13 14:33:45 +01:00
code_llama
codegen Generation tests: update imagegpt input name, remove unused functions (#33663) 2024-09-24 16:40:48 +01:00
cohere Add Cohere2 model (#35224) 2024-12-13 09:35:50 +01:00
cohere2 Refactor/fix Cohere2 (#35594) 2025-01-09 17:54:57 +01:00
colpali Add ColPali to 🤗 transformers (#33736) 2024-12-17 11:26:43 +01:00
conditional_detr [tests] fix "Tester object has no attribute '_testMethodName'" (#34910) 2024-12-13 14:33:45 +01:00
convbert
convnext Update kwargs validation for preprocess with decorator (#32024) 2024-08-06 11:33:05 +01:00
convnextv2
cpm Make pipeline able to load processor (#32514) 2024-10-09 16:46:11 +01:00
cpmant
ctrl avoid calling gc.collect and cuda.empty_cache (#34514) 2024-10-31 16:36:13 +01:00
cvt
dac [tests] fix "Tester object has no attribute '_testMethodName'" (#34910) 2024-12-13 14:33:45 +01:00
data2vec Add sdpa for Beit (#34941) 2024-12-17 14:44:47 +01:00
dbrx [Compile] Only test compiling model forward pass (#35658) 2025-01-13 13:43:29 +01:00
deberta [Deberta/Deberta-v2] Refactor code base to support compile, export, and fix LLM (#22105) 2024-11-25 10:43:16 +01:00
deberta_v2 [Deberta/Deberta-v2] Refactor code base to support compile, export, and fix LLM (#22105) 2024-11-25 10:43:16 +01:00
decision_transformer Forbid PretrainedConfig from saving generate parameters; Update deprecations in generate-related code 🧹 (#32659) 2024-08-23 11:12:53 +01:00
deformable_detr Add Image Processor Fast Deformable DETR (#34353) 2024-11-19 11:18:58 -05:00
deit Update kwargs validation for preprocess with decorator (#32024) 2024-08-06 11:33:05 +01:00
depth_anything fix(DPT,Depth-Anything) torch.export (#34103) 2024-11-20 11:31:21 +01:00
detr Fall back to slow image processor in ImageProcessingAuto when no fast processor available (#34785) 2024-12-15 14:00:36 -05:00
diffllama [tests] make cuda-only tests device-agnostic (#35607) 2025-01-13 14:48:39 +01:00
dinat
dinov2 Sdpa dino v2 (#33403) 2024-09-21 01:58:00 +01:00
dinov2_with_registers Add DINOv2 with registers (#35348) 2024-12-24 13:21:59 +01:00
distilbert DistilBERT is ExecuTorch compatible (#34475) 2024-11-05 13:41:48 +01:00
dit
donut rename all test_processing_*.py to test_processor_*.py (#33878) 2024-10-02 16:43:43 +02:00
dpr
dpt fix(DPT,Depth-Anything) torch.export (#34103) 2024-11-20 11:31:21 +01:00
efficientnet CI: fix efficientnet pipeline timeout and prevent future similar issues due to large image size (#33123) 2024-08-27 11:58:27 +01:00
electra [pytes collection] Fix flax test collection (#34004) 2024-10-07 18:11:13 +02:00
emu3 [Compile] Only test compiling model forward pass (#35658) 2025-01-13 13:43:29 +01:00
encodec Fix bug when requesting input normalization with EnCodec (#34756) 2025-01-07 11:50:02 +01:00
encoder_decoder 🚨All attention refactor🚨 (#35235) 2024-12-18 16:53:39 +01:00
ernie
esm
falcon 🚨All attention refactor🚨 (#35235) 2024-12-18 16:53:39 +01:00
falcon_mamba [tests] make cuda-only tests device-agnostic (#35607) 2025-01-13 14:48:39 +01:00
fastspeech2_conformer
flaubert Make pipeline able to load processor (#32514) 2024-10-09 16:46:11 +01:00
flava Load sub-configs from composite configs (#34410) 2024-11-05 11:34:01 +01:00
fnet Make pipeline able to load processor (#32514) 2024-10-09 16:46:11 +01:00
focalnet
fsmt
funnel
fuyu [tests] make cuda-only tests device-agnostic (#35607) 2025-01-13 14:48:39 +01:00
gemma Tests: move generate tests to the right mixin and delete redundant tests (#34464) 2024-10-30 10:59:08 +00:00
gemma2 [FlexAttention] Update gemma2 (#34942) 2024-11-27 11:50:48 +01:00
git Add image text to text pipeline (#34170) 2024-10-31 15:48:11 -04:00
glm Remove @slow for test_eager_matches_sdpa_inference (#34558) 2024-11-05 16:10:42 +01:00
glpn Add post_process_depth_estimation for GLPN (#34413) 2024-10-28 19:44:20 +01:00
gpt2 🚨All attention refactor🚨 (#35235) 2024-12-18 16:53:39 +01:00
gpt_bigcode Aurevoir PyTorch 1 (#35358) 2024-12-20 14:36:31 +01:00
gpt_neo Generation tests: update imagegpt input name, remove unused functions (#33663) 2024-09-24 16:40:48 +01:00
gpt_neox 🚨All attention refactor🚨 (#35235) 2024-12-18 16:53:39 +01:00
gpt_neox_japanese Compile compatibilty for decoder-only models (#32617) 2024-09-09 10:59:04 +02:00
gpt_sw3 🚨 No more default chat templates (#31733) 2024-07-24 17:36:32 +01:00
gptj Aurevoir PyTorch 1 (#35358) 2024-12-20 14:36:31 +01:00
granite Fix some fa2 tests (#35340) 2024-12-19 17:05:25 +01:00
granitemoe Fix some fa2 tests (#35340) 2024-12-19 17:05:25 +01:00
grounding_dino [tests] fix "Tester object has no attribute '_testMethodName'" (#34910) 2024-12-13 14:33:45 +01:00
groupvit Load sub-configs from composite configs (#34410) 2024-11-05 11:34:01 +01:00
helium Add-helium (#35669) 2025-01-13 18:41:15 +01:00
herbert
hiera 🚨 Fix torch.jit.trace for interpolate_pos_encoding in all vision models (#33226) 2024-09-05 16:17:34 +02:00
hubert Support BatchNorm in Hubert pos_conv_emb as in fairseq (#34389) 2024-12-10 14:18:23 +01:00
ibert [warnings] fix E721 warnings (#32223) 2024-07-25 15:12:23 +02:00
idefics [Compile] Only test compiling model forward pass (#35658) 2025-01-13 13:43:29 +01:00
idefics2 Aurevoir PyTorch 1 (#35358) 2024-12-20 14:36:31 +01:00
idefics3 bugfix Idefics3 processor - handle gracefully cases with text and no images (#35363) 2024-12-23 16:59:01 +01:00
ijepa Update I-JEPA checkpoints path (#35120) 2024-12-06 13:42:51 +00:00
imagegpt Simplify running tests in a subprocess (#34213) 2024-10-29 10:48:57 +01:00
informer tests: fix pytorch tensor placement errors (#33485) 2024-09-25 12:21:53 +01:00
instructblip BLIP: fix generation after hub update (#34876) 2024-11-25 10:41:55 +01:00
instructblipvideo Load sub-configs from composite configs (#34410) 2024-11-05 11:34:01 +01:00
jamba Tests: move generate tests to the right mixin and delete redundant tests (#34464) 2024-10-30 10:59:08 +00:00
jetmoe Tests: move generate tests to the right mixin and delete redundant tests (#34464) 2024-10-30 10:59:08 +00:00
kosmos2 VLMs: enable generation tests - last batch (#34484) 2024-11-21 11:00:22 +01:00
layoutlm
layoutlmv2 Fix #34494 assistant tokens when truncated (#34531) 2024-11-05 15:10:15 +00:00
layoutlmv3 Fix #34494 assistant tokens when truncated (#34531) 2024-11-05 15:10:15 +00:00
layoutxlm Fix #34494 assistant tokens when truncated (#34531) 2024-11-05 15:10:15 +00:00
led Make pipeline able to load processor (#32514) 2024-10-09 16:46:11 +01:00
levit Update kwargs validation for preprocess with decorator (#32024) 2024-08-06 11:33:05 +01:00
lilt Make pipeline able to load processor (#32514) 2024-10-09 16:46:11 +01:00
llama [tests] make cuda-only tests device-agnostic (#35607) 2025-01-13 14:48:39 +01:00
llava Chat template: return vectorized output in processors (#34275) 2025-01-10 11:05:29 +01:00
llava_next VLMs: major clean up 🧼 (#34502) 2025-01-08 10:35:23 +01:00
llava_next_video VLMs: major clean up 🧼 (#34502) 2025-01-08 10:35:23 +01:00
llava_onevision Chat template: return vectorized output in processors (#34275) 2025-01-10 11:05:29 +01:00
longformer Make pipeline able to load processor (#32514) 2024-10-09 16:46:11 +01:00
longt5 T5 compile compatibilty (#34089) 2024-10-22 08:23:53 +02:00
luke Make pipeline able to load processor (#32514) 2024-10-09 16:46:11 +01:00
lxmert Refactor: Removed un-necessary object base class (#32230) 2024-07-26 10:33:02 +02:00
m2m_100 Make pipeline able to load processor (#32514) 2024-10-09 16:46:11 +01:00
mamba Aurevoir PyTorch 1 (#35358) 2024-12-20 14:36:31 +01:00
mamba2 Aurevoir PyTorch 1 (#35358) 2024-12-20 14:36:31 +01:00
marian Forbid PretrainedConfig from saving generate parameters; Update deprecations in generate-related code 🧹 (#32659) 2024-08-23 11:12:53 +01:00
markuplm [tests] fix "Tester object has no attribute '_testMethodName'" (#34910) 2024-12-13 14:33:45 +01:00
mask2former [tests] fix "Tester object has no attribute '_testMethodName'" (#34910) 2024-12-13 14:33:45 +01:00
maskformer [tests] fix "Tester object has no attribute '_testMethodName'" (#34910) 2024-12-13 14:33:45 +01:00
mbart Make pipeline able to load processor (#32514) 2024-10-09 16:46:11 +01:00
mbart50
megatron_bert
megatron_gpt2
mgp_str Remove repeated prepare_images in processor tests (#33163) 2024-09-09 13:20:27 +01:00
mimi Fix test_eager_matches_sdpa_inference for XPU backend (#34889) 2024-12-02 16:21:04 +01:00
mistral [tests] make cuda-only tests device-agnostic (#35607) 2025-01-13 14:48:39 +01:00
mixtral [tests] make cuda-only tests device-agnostic (#35607) 2025-01-13 14:48:39 +01:00
mllama Chat template: return vectorized output in processors (#34275) 2025-01-10 11:05:29 +01:00
mluke fix: Updated test_embeded_special_tokens for luke and mluke models (#32413) 2024-08-05 15:19:42 +01:00
mobilebert MobileBERT is ExecuTorch compatible (#34473) 2024-10-29 16:14:31 +01:00
mobilenet_v1 Skip MobileNetV1ModelTest::test_batching_equivalence for now (#35614) 2025-01-10 18:32:36 +01:00
mobilenet_v2 Update kwargs validation for preprocess with decorator (#32024) 2024-08-06 11:33:05 +01:00
mobilevit Forbid PretrainedConfig from saving generate parameters; Update deprecations in generate-related code 🧹 (#32659) 2024-08-23 11:12:53 +01:00
mobilevitv2
modernbert ModernBert: reuse GemmaRotaryEmbedding via modular + Integration tests (#35459) 2025-01-10 10:25:10 +01:00
moonshine Add Moonshine (#34784) 2025-01-10 11:00:54 +01:00
moshi VLM: special multimodal Tokenizer (#34461) 2024-11-04 16:37:51 +01:00
mpnet
mpt
mra Fix Failed tests with mobile bert resize tokens embedding (#33950) 2024-10-09 11:23:50 +01:00
mt5 Tests: move generate tests to the right mixin and delete redundant tests (#34464) 2024-10-30 10:59:08 +00:00
musicgen Fix test_eager_matches_sdpa_inference for XPU backend (#34889) 2024-12-02 16:21:04 +01:00
musicgen_melody Make test_generate_with_static_cache even less flaky (#34995) 2024-12-20 16:03:26 +01:00
mvp Make pipeline able to load processor (#32514) 2024-10-09 16:46:11 +01:00
myt5 [Red CIs] Fix hub failures (#34001) 2024-10-07 10:56:24 +02:00
nemotron [tests] make cuda-only tests device-agnostic (#35607) 2025-01-13 14:48:39 +01:00
nllb
nllb_moe Make pipeline able to load processor (#32514) 2024-10-09 16:46:11 +01:00
nougat Update kwargs validation for preprocess with decorator (#32024) 2024-08-06 11:33:05 +01:00
nystromformer
olmo Tests: upgrade test_eager_matches_sdpa_generate (#34386) 2024-10-25 11:55:07 +01:00
olmo2 Rename OLMo November to OLMo2 (#34864) 2024-11-25 16:31:22 +01:00
olmoe Tests: upgrade test_eager_matches_sdpa_generate (#34386) 2024-10-25 11:55:07 +01:00
omdet_turbo [tests] make cuda-only tests device-agnostic (#35607) 2025-01-13 14:48:39 +01:00
oneformer [tests] fix "Tester object has no attribute '_testMethodName'" (#34910) 2024-12-13 14:33:45 +01:00
openai Make pipeline able to load processor (#32514) 2024-10-09 16:46:11 +01:00
opt Tests: upgrade test_eager_matches_sdpa_generate (#34386) 2024-10-25 11:55:07 +01:00
owlv2 owlvit/2 dynamic input resolution (#34764) 2024-12-21 08:51:09 +00:00
owlvit owlvit/2 dynamic input resolution (#34764) 2024-12-21 08:51:09 +00:00
paligemma [Compile] Only test compiling model forward pass (#35658) 2025-01-13 13:43:29 +01:00
patchtsmixer
patchtst
pegasus Forbid PretrainedConfig from saving generate parameters; Update deprecations in generate-related code 🧹 (#32659) 2024-08-23 11:12:53 +01:00
pegasus_x
perceiver
persimmon 🚨All attention refactor🚨 (#35235) 2024-12-18 16:53:39 +01:00
phi 🚨All attention refactor🚨 (#35235) 2024-12-18 16:53:39 +01:00
phi3 More model refactoring! (#35359) 2025-01-09 11:09:09 +01:00
phimoe fix red check-copies (#33964) 2024-10-04 22:45:37 +02:00
phobert
pix2struct [tests] fix "Tester object has no attribute '_testMethodName'" (#34910) 2024-12-13 14:33:45 +01:00
pixtral [PixtralLarge] Update Pixtral conversion script to support large format! (#34801) 2025-01-08 17:39:47 +01:00
plbart Make pipeline able to load processor (#32514) 2024-10-09 16:46:11 +01:00
poolformer Update kwargs validation for preprocess with decorator (#32024) 2024-08-06 11:33:05 +01:00
pop2piano [tests] fix "Tester object has no attribute '_testMethodName'" (#34910) 2024-12-13 14:33:45 +01:00
prophetnet Make pipeline able to load processor (#32514) 2024-10-09 16:46:11 +01:00
pvt Update kwargs validation for preprocess with decorator (#32024) 2024-08-06 11:33:05 +01:00
pvt_v2
qwen2 🚨All attention refactor🚨 (#35235) 2024-12-18 16:53:39 +01:00
qwen2_audio [Qwen2Audio] handle input ids expansion during processing (#35534) 2025-01-07 16:47:27 +01:00
qwen2_moe 🚨All attention refactor🚨 (#35235) 2024-12-18 16:53:39 +01:00
qwen2_vl [Compile] Only test compiling model forward pass (#35658) 2025-01-13 13:43:29 +01:00
rag [tests] make cuda-only tests device-agnostic (#35222) 2024-12-18 10:14:22 +01:00
recurrent_gemma Tests: move generate tests to the right mixin and delete redundant tests (#34464) 2024-10-30 10:59:08 +00:00
reformer Generation tests: don't rely on main input name (#34228) 2024-10-21 10:00:14 +02:00
regnet [pytes collection] Fix flax test collection (#34004) 2024-10-07 18:11:13 +02:00
rembert VLM: special multimodal Tokenizer (#34461) 2024-11-04 16:37:51 +01:00
resnet [pytes collection] Fix flax test collection (#34004) 2024-10-07 18:11:13 +02:00
roberta Roberta is ExecuTorch compatible (#34425) 2024-10-30 08:36:45 +00:00
roberta_prelayernorm [pytes collection] Fix flax test collection (#34004) 2024-10-07 18:11:13 +02:00
roc_bert Make pipeline able to load processor (#32514) 2024-10-09 16:46:11 +01:00
roformer Make pipeline able to load processor (#32514) 2024-10-09 16:46:11 +01:00
rt_detr [tests] make cuda-only tests device-agnostic (#35607) 2025-01-13 14:48:39 +01:00
rwkv Aurevoir PyTorch 1 (#35358) 2024-12-20 14:36:31 +01:00
sam uniformize kwargs for SAM (#34578) 2024-12-23 13:54:57 +01:00
seamless_m4t [tests] fix "Tester object has no attribute '_testMethodName'" (#34910) 2024-12-13 14:33:45 +01:00
seamless_m4t_v2 Make test_generate_with_static_cache even less flaky (#34995) 2024-12-20 16:03:26 +01:00
segformer [tests] fix "Tester object has no attribute '_testMethodName'" (#34910) 2024-12-13 14:33:45 +01:00
seggpt [tests] fix "Tester object has no attribute '_testMethodName'" (#34910) 2024-12-13 14:33:45 +01:00
sew Fix parametrization-based weight norm (#33275) 2024-09-17 08:05:21 +02:00
sew_d Fix parametrization-based weight norm (#33275) 2024-09-17 08:05:21 +02:00
siglip Load sub-configs from composite configs (#34410) 2024-11-05 11:34:01 +01:00
speech_encoder_decoder 🚨All attention refactor🚨 (#35235) 2024-12-18 16:53:39 +01:00
speech_to_text [tests] fix "Tester object has no attribute '_testMethodName'" (#34910) 2024-12-13 14:33:45 +01:00
speecht5 [tests] fix "Tester object has no attribute '_testMethodName'" (#34910) 2024-12-13 14:33:45 +01:00
splinter Make pipeline able to load processor (#32514) 2024-10-09 16:46:11 +01:00
squeezebert Refactor: Removed un-necessary object base class (#32230) 2024-07-26 10:33:02 +02:00
stablelm 🚨All attention refactor🚨 (#35235) 2024-12-18 16:53:39 +01:00
starcoder2 [tests] make cuda-only tests device-agnostic (#35607) 2025-01-13 14:48:39 +01:00
superpoint [tests] fix "Tester object has no attribute '_testMethodName'" (#34910) 2024-12-13 14:33:45 +01:00
swiftformer
swin
swin2sr Update kwargs validation for preprocess with decorator (#32024) 2024-08-06 11:33:05 +01:00
swinv2
switch_transformers Fix flaky SwitchTransformersModelTest::test_training_gradient (#35587) 2025-01-09 15:36:22 +01:00
t5 [tests] make cuda-only tests device-agnostic (#35607) 2025-01-13 14:48:39 +01:00
table_transformer
tapas Aurevoir PyTorch 1 (#35358) 2024-12-20 14:36:31 +01:00
textnet Add TextNet (#34979) 2025-01-08 09:52:51 +01:00
time_series_transformer
timesformer
timm_backbone
timm_wrapper Enable auto task for timm models in pipeline (#35531) 2025-01-08 15:14:17 +00:00
trocr 🚨🚨🚨 Uniformize kwargs for TrOCR Processor (#34587) 2024-11-29 11:58:11 +00:00
tvp Update kwargs validation for preprocess with decorator (#32024) 2024-08-06 11:33:05 +01:00
udop Fix #34494 assistant tokens when truncated (#34531) 2024-11-05 15:10:15 +00:00
umt5 Tests: move generate tests to the right mixin and delete redundant tests (#34464) 2024-10-30 10:59:08 +00:00
unispeech Remove trust_remote_code when loading Libri Dummy (#31748) 2024-07-23 14:54:38 +08:00
unispeech_sat Remove trust_remote_code when loading Libri Dummy (#31748) 2024-07-23 14:54:38 +08:00
univnet [tests] fix "Tester object has no attribute '_testMethodName'" (#34910) 2024-12-13 14:33:45 +01:00
upernet Fix flaky test_batching_equivalence (#35564) 2025-01-09 14:00:08 +01:00
video_llava VLMs: major clean up 🧼 (#34502) 2025-01-08 10:35:23 +01:00
videomae Remove @slow for test_eager_matches_sdpa_inference (#34558) 2024-11-05 16:10:42 +01:00
vilt Update kwargs validation for preprocess with decorator (#32024) 2024-08-06 11:33:05 +01:00
vipllava VLMs: major clean up 🧼 (#34502) 2025-01-08 10:35:23 +01:00
vision_encoder_decoder 🚨All attention refactor🚨 (#35235) 2024-12-18 16:53:39 +01:00
vision_text_dual_encoder Fall back to slow image processor in ImageProcessingAuto when no fast processor available (#34785) 2024-12-15 14:00:36 -05:00
visual_bert
vit [pytes collection] Fix flax test collection (#34004) 2024-10-07 18:11:13 +02:00
vit_mae Fix ViT-MAE decoder interpolate (#33330) 2024-09-30 18:47:13 +02:00
vit_msn
vitdet
vitmatte Update kwargs validation for preprocess with decorator (#32024) 2024-08-06 11:33:05 +01:00
vitpose Add ViTPose (#30530) 2025-01-08 16:02:14 +00:00
vitpose_backbone Add ViTPose (#30530) 2025-01-08 16:02:14 +00:00
vits Generate tests: modality-agnostic input preparation (#33685) 2024-10-03 14:01:24 +01:00
vivit Add sdpa for Vivit (#33757) 2024-10-15 11:27:54 +02:00
wav2vec2 [tests] fix "Tester object has no attribute '_testMethodName'" (#34910) 2024-12-13 14:33:45 +01:00
wav2vec2_bert Uniformize model processors (#31368) 2024-10-02 10:41:08 +02:00
wav2vec2_conformer Remove trust_remote_code when loading Libri Dummy (#31748) 2024-07-23 14:54:38 +08:00
wav2vec2_phoneme clean_up_tokenization_spaces=False if unset (#31938) 2024-09-26 19:38:20 +02:00
wav2vec2_with_lm
wavlm Remove trust_remote_code when loading Libri Dummy (#31748) 2024-07-23 14:54:38 +08:00
whisper [Whisper] 🚨 Fix whisper decoding 🚨 (#34135) 2024-12-18 14:13:21 +01:00
x_clip Load sub-configs from composite configs (#34410) 2024-11-05 11:34:01 +01:00
xglm avoid calling gc.collect and cuda.empty_cache (#34514) 2024-10-31 16:36:13 +01:00
xlm Make pipeline able to load processor (#32514) 2024-10-09 16:46:11 +01:00
xlm_roberta [RoBERTa-based] Add support for sdpa (#30510) 2024-08-28 10:26:00 +02:00
xlm_roberta_xl Tests: upgrade test_eager_matches_sdpa_generate (#34386) 2024-10-25 11:55:07 +01:00
xlnet Make pipeline able to load processor (#32514) 2024-10-09 16:46:11 +01:00
xmod Make pipeline able to load processor (#32514) 2024-10-09 16:46:11 +01:00
yolos [tests] fix "Tester object has no attribute '_testMethodName'" (#34910) 2024-12-13 14:33:45 +01:00
yoso
zamba Zamba new attention standard (#35375) 2025-01-07 10:08:45 +01:00
zoedepth Add post_process_depth_estimation to image processors and support ZoeDepth's inference intricacies (#32550) 2024-10-22 15:50:54 +02:00
__init__.py