transformers/tests/models
Raushan Turganbay 913330ca9f
VLMs: fix number of image tokens (#34332)
* fix

* fix tests

* add tests

* style

* style

* fix qwen after rebase

* fix video llava
2024-10-30 10:21:37 +01:00
..
albert Albert is ExecuTorch compatible (#34476) 2024-10-29 16:22:13 +01:00
align Uniformize kwargs for image-text-to-text processors (#32544) 2024-09-24 21:28:19 -04:00
altclip Make pipeline able to load processor (#32514) 2024-10-09 16:46:11 +01:00
audio_spectrogram_transformer Make pipeline able to load processor (#32514) 2024-10-09 16:46:11 +01:00
auto Generation: deprecate PreTrainedModel inheriting from GenerationMixin (#33203) 2024-09-23 18:28:36 +01:00
autoformer Rename test_model_common_attributes -> test_model_get_set_embeddings (#31321) 2024-06-07 19:40:26 +01:00
bark Skip tests properly (#31308) 2024-06-26 21:59:08 +01:00
bart Forbid PretrainedConfig from saving generate parameters; Update deprecations in generate-related code 🧹 (#32659) 2024-08-23 11:12:53 +01:00
barthez Skip tests properly (#31308) 2024-06-26 21:59:08 +01:00
bartpho Adds pretrained IDs directly in the tests (#29534) 2024-03-13 14:53:27 +01:00
beit [pytes collection] Fix flax test collection (#34004) 2024-10-07 18:11:13 +02:00
bert Bert is ExecuTorch compatible (#34424) 2024-10-29 14:30:02 +01:00
bert_generation Adds pretrained IDs directly in the tests (#29534) 2024-03-13 14:53:27 +01:00
bert_japanese Adds pretrained IDs directly in the tests (#29534) 2024-03-13 14:53:27 +01:00
bertweet Adds pretrained IDs directly in the tests (#29534) 2024-03-13 14:53:27 +01:00
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 Use self.config_tester.run_common_tests() (#31431) 2024-06-19 10:18:08 +01:00
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 rename all test_processing_*.py to test_processor_*.py (#33878) 2024-10-02 16:43:43 +02:00
blip_2 Attn implementation for composite models (#32238) 2024-10-22 06:54:44 +02: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 Skip tests properly (#31308) 2024-06-26 21:59:08 +01:00
camembert [RoBERTa-based] Add support for sdpa (#30510) 2024-08-28 10:26:00 +02:00
canine Skip tests properly (#31308) 2024-06-26 21:59:08 +01:00
chameleon Generate tests: modality-agnostic input preparation (#33685) 2024-10-03 14:01:24 +01:00
chinese_clip adding positional encoder changes and tests (#32600) 2024-09-25 19:05:01 +01:00
clap Remove trust_remote_code when loading Libri Dummy (#31748) 2024-07-23 14:54:38 +08:00
clip Attn implementation for composite models (#32238) 2024-10-22 06:54:44 +02:00
clipseg adding positional encoder changes and tests (#32600) 2024-09-25 19:05:01 +01:00
clvp clean_up_tokenization_spaces=False if unset (#31938) 2024-09-26 19:38:20 +02:00
code_llama Skip tests properly (#31308) 2024-06-26 21:59:08 +01:00
codegen Generation tests: update imagegpt input name, remove unused functions (#33663) 2024-09-24 16:40:48 +01:00
cohere Tests: upgrade test_eager_matches_sdpa_generate (#34386) 2024-10-25 11:55:07 +01:00
conditional_detr Add DetrImageProcessorFast (#34063) 2024-10-21 09:05:05 -04:00
convbert Skip tests properly (#31308) 2024-06-26 21:59:08 +01:00
convnext Update kwargs validation for preprocess with decorator (#32024) 2024-08-06 11:33:05 +01:00
convnextv2 Skip tests properly (#31308) 2024-06-26 21:59:08 +01:00
cpm Make pipeline able to load processor (#32514) 2024-10-09 16:46:11 +01:00
cpmant Skip tests properly (#31308) 2024-06-26 21:59:08 +01:00
ctrl Make pipeline able to load processor (#32514) 2024-10-09 16:46:11 +01:00
cvt Use self.config_tester.run_common_tests() (#31431) 2024-06-19 10:18:08 +01:00
dac Fix DAC slow tests (#34088) 2024-10-11 14:43:03 +02:00
data2vec Remove trust_remote_code when loading Libri Dummy (#31748) 2024-07-23 14:54:38 +08:00
dbrx Generate: end-to-end compilation (#30788) 2024-07-29 10:52:13 +01:00
deberta Refactor: Removed un-necessary object base class (#32230) 2024-07-26 10:33:02 +02:00
deberta_v2 Refactor: Removed un-necessary object base class (#32230) 2024-07-26 10:33:02 +02: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 DetrImageProcessorFast (#34063) 2024-10-21 09:05:05 -04:00
deit Update kwargs validation for preprocess with decorator (#32024) 2024-08-06 11:33:05 +01:00
depth_anything Add Depth Anything V2 Metric models (#32126) 2024-08-13 16:16:30 +02:00
detr Add DetrImageProcessorFast (#34063) 2024-10-21 09:05:05 -04:00
dinat Skip tests properly (#31308) 2024-06-26 21:59:08 +01:00
dinov2 Sdpa dino v2 (#33403) 2024-09-21 01:58:00 +01:00
distilbert [pytes collection] Fix flax test collection (#34004) 2024-10-07 18:11:13 +02:00
dit Update old existing feature extractor references (#24552) 2023-06-29 10:17:36 +01:00
donut rename all test_processing_*.py to test_processor_*.py (#33878) 2024-10-02 16:43:43 +02:00
dpr Remove static pretrained maps from the library's internals (#29112) 2024-03-25 10:33:38 +01:00
dpt Add post_process_depth_estimation to image processors and support ZoeDepth's inference intricacies (#32550) 2024-10-22 15:50:54 +02: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
encodec Generate tests: modality-agnostic input preparation (#33685) 2024-10-03 14:01:24 +01:00
encoder_decoder Correct the new defaults (#34377) 2024-10-24 18:42:03 +02:00
ernie Skip tests properly (#31308) 2024-06-26 21:59:08 +01:00
esm Skip tests properly (#31308) 2024-06-26 21:59:08 +01:00
falcon Tests: upgrade test_eager_matches_sdpa_generate (#34386) 2024-10-25 11:55:07 +01:00
falcon_mamba Generation tests: update imagegpt input name, remove unused functions (#33663) 2024-09-24 16:40:48 +01:00
fastspeech2_conformer Skip tests properly (#31308) 2024-06-26 21:59:08 +01:00
flaubert Make pipeline able to load processor (#32514) 2024-10-09 16:46:11 +01:00
flava Remove repeated prepare_images in processor tests (#33163) 2024-09-09 13:20:27 +01:00
fnet Make pipeline able to load processor (#32514) 2024-10-09 16:46:11 +01:00
focalnet Use self.config_tester.run_common_tests() (#31431) 2024-06-19 10:18:08 +01:00
fsmt Skip tests properly (#31308) 2024-06-26 21:59:08 +01:00
funnel update ruff version (#30932) 2024-05-22 06:40:15 +02:00
fuyu rename all test_processing_*.py to test_processor_*.py (#33878) 2024-10-02 16:43:43 +02:00
gemma Generate using exported model and enable gemma2-2b in ExecuTorch (#33707) 2024-10-11 10:16:31 +02:00
gemma2 Attn implementation for composite models (#32238) 2024-10-22 06:54:44 +02:00
git adding positional encoder changes and tests (#32600) 2024-09-25 19:05:01 +01:00
glm Tests: upgrade test_eager_matches_sdpa_generate (#34386) 2024-10-25 11:55:07 +01:00
glpn Add post_process_depth_estimation for GLPN (#34413) 2024-10-28 19:44:20 +01:00
gpt_bigcode Generation tests: update imagegpt input name, remove unused functions (#33663) 2024-09-24 16:40:48 +01:00
gpt_neo Generation tests: update imagegpt input name, remove unused functions (#33663) 2024-09-24 16:40:48 +01:00
gpt_neox Tests: upgrade test_eager_matches_sdpa_generate (#34386) 2024-10-25 11:55:07 +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
gpt2 🚨 No more default chat templates (#31733) 2024-07-24 17:36:32 +01:00
gptj Make pipeline able to load processor (#32514) 2024-10-09 16:46:11 +01:00
granite Tests: upcast logits to float() (#34042) 2024-10-11 11:51:49 +01:00
granitemoe Tests: upcast logits to float() (#34042) 2024-10-11 11:51:49 +01:00
grounding_dino Add DetrImageProcessorFast (#34063) 2024-10-21 09:05:05 -04:00
groupvit Skip tests properly (#31308) 2024-06-26 21:59:08 +01:00
herbert Skip tests properly (#31308) 2024-06-26 21:59:08 +01:00
hiera 🚨 Fix torch.jit.trace for interpolate_pos_encoding in all vision models (#33226) 2024-09-05 16:17:34 +02:00
hubert Remove trust_remote_code when loading Libri Dummy (#31748) 2024-07-23 14:54:38 +08:00
ibert [warnings] fix E721 warnings (#32223) 2024-07-25 15:12:23 +02:00
idefics Generation: fix test (#34369) 2024-10-29 07:57:10 +01:00
idefics2 Use non nested images and batched text Idefics2/3 (#34222) 2024-10-24 20:00:13 -04:00
idefics3 Use non nested images and batched text Idefics2/3 (#34222) 2024-10-24 20:00:13 -04: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 CI: fix failures (#34371) 2024-10-24 13:44:53 +02:00
instructblipvideo Attn implementation for composite models (#32238) 2024-10-22 06:54:44 +02:00
jamba add test for Jamba with new model jamba-tiny-dev (#33863) 2024-10-05 16:03:12 +02:00
jetmoe Tests: upgrade test_eager_matches_sdpa_generate (#34386) 2024-10-25 11:55:07 +01:00
kosmos2 Attn implementation for composite models (#32238) 2024-10-22 06:54:44 +02:00
layoutlm Skip tests properly (#31308) 2024-06-26 21:59:08 +01:00
layoutlmv2 [tests] skip tests for xpu (#33553) 2024-09-19 19:28:04 +01:00
layoutlmv3 Make pipeline able to load processor (#32514) 2024-10-09 16:46:11 +01:00
layoutxlm Enable padding_side as call time kwargs (#33385) 2024-09-13 11:58:38 +01: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: upgrade test_eager_matches_sdpa_generate (#34386) 2024-10-25 11:55:07 +01:00
llava VLMs: fix number of image tokens (#34332) 2024-10-30 10:21:37 +01:00
llava_next VLMs: fix number of image tokens (#34332) 2024-10-30 10:21:37 +01:00
llava_next_video VLMs: fix number of image tokens (#34332) 2024-10-30 10:21:37 +01:00
llava_onevision Attn implementation for composite models (#32238) 2024-10-22 06:54:44 +02: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 Fix Mamba slow path bug with dtype mismatch. (#32691) 2024-10-01 09:28:40 +02:00
mamba2 Generation: fix test (#34369) 2024-10-29 07:57:10 +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 Make pipeline able to load processor (#32514) 2024-10-09 16:46:11 +01:00
mask2former fixed Mask2Former image processor segmentation maps handling (#33364) 2024-09-10 11:19:56 +01:00
maskformer Skip tests properly (#31308) 2024-06-26 21:59:08 +01:00
mbart Make pipeline able to load processor (#32514) 2024-10-09 16:46:11 +01:00
mbart50 Skip tests properly (#31308) 2024-06-26 21:59:08 +01:00
megatron_bert Skip tests properly (#31308) 2024-06-26 21:59:08 +01:00
megatron_gpt2 Skip tests properly (#31308) 2024-06-26 21:59:08 +01:00
mgp_str Remove repeated prepare_images in processor tests (#33163) 2024-09-09 13:20:27 +01:00
mimi Moshi integration (#33624) 2024-10-16 11:21:49 +02:00
mistral Tests: upgrade test_eager_matches_sdpa_generate (#34386) 2024-10-25 11:55:07 +01:00
mixtral Tests: upgrade test_eager_matches_sdpa_generate (#34386) 2024-10-25 11:55:07 +01:00
mllama Tests: upgrade test_eager_matches_sdpa_generate (#34386) 2024-10-25 11:55:07 +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 Update kwargs validation for preprocess with decorator (#32024) 2024-08-06 11:33:05 +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 Rename test_model_common_attributes -> test_model_get_set_embeddings (#31321) 2024-06-07 19:40:26 +01:00
moshi Generation: fix test (#34369) 2024-10-29 07:57:10 +01:00
mpnet Skip tests properly (#31308) 2024-06-26 21:59:08 +01:00
mpt Skip tests properly (#31308) 2024-06-26 21:59:08 +01:00
mra Fix Failed tests with mobile bert resize tokens embedding (#33950) 2024-10-09 11:23:50 +01:00
mt5 T5 compile compatibilty (#34089) 2024-10-22 08:23:53 +02:00
musicgen Tests: upgrade test_eager_matches_sdpa_generate (#34386) 2024-10-25 11:55:07 +01:00
musicgen_melody Tests: upgrade test_eager_matches_sdpa_generate (#34386) 2024-10-25 11:55:07 +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 Fix attn mask ignore logic in training-time trace (#32613) 2024-10-04 19:00:45 +02:00
nllb Skip tests properly (#31308) 2024-06-26 21:59:08 +01:00
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 update ruff version (#30932) 2024-05-22 06:40:15 +02:00
olmo Tests: upgrade test_eager_matches_sdpa_generate (#34386) 2024-10-25 11:55:07 +01:00
olmoe Tests: upgrade test_eager_matches_sdpa_generate (#34386) 2024-10-25 11:55:07 +01:00
omdet_turbo Add support for custom inputs and batched inputs in ProcessorTesterMixin (#33711) 2024-10-01 23:52:03 +02:00
oneformer Make pipeline able to load processor (#32514) 2024-10-09 16:46:11 +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 Update kwargs validation for preprocess with decorator (#32024) 2024-08-06 11:33:05 +01:00
owlvit Remove repeated prepare_images in processor tests (#33163) 2024-09-09 13:20:27 +01:00
paligemma VLMs: fix number of image tokens (#34332) 2024-10-30 10:21:37 +01:00
patchtsmixer Skip tests properly (#31308) 2024-06-26 21:59:08 +01:00
patchtst Rename test_model_common_attributes -> test_model_get_set_embeddings (#31321) 2024-06-07 19:40:26 +01:00
pegasus Forbid PretrainedConfig from saving generate parameters; Update deprecations in generate-related code 🧹 (#32659) 2024-08-23 11:12:53 +01:00
pegasus_x Skip tests properly (#31308) 2024-06-26 21:59:08 +01:00
perceiver Skip tests properly (#31308) 2024-06-26 21:59:08 +01:00
persimmon Tests: upcast logits to float() (#34042) 2024-10-11 11:51:49 +01:00
phi Make pipeline able to load processor (#32514) 2024-10-09 16:46:11 +01:00
phi3 Phi3: fix attn for sliding window (#33586) 2024-10-10 11:50:39 +02:00
phimoe fix red check-copies (#33964) 2024-10-04 22:45:37 +02:00
phobert Adds pretrained IDs directly in the tests (#29534) 2024-03-13 14:53:27 +01:00
pix2struct Fix pix2struct (#34374) 2024-10-28 11:24:56 +01:00
pixtral Add support for custom inputs and batched inputs in ProcessorTesterMixin (#33711) 2024-10-01 23:52:03 +02: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 T5 compile compatibilty (#34089) 2024-10-22 08:23:53 +02: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 Skip tests properly (#31308) 2024-06-26 21:59:08 +01:00
qwen2 Tests: upgrade test_eager_matches_sdpa_generate (#34386) 2024-10-25 11:55:07 +01:00
qwen2_audio Attn implementation for composite models (#32238) 2024-10-22 06:54:44 +02:00
qwen2_moe Tests: upgrade test_eager_matches_sdpa_generate (#34386) 2024-10-25 11:55:07 +01:00
qwen2_vl VLMs: fix number of image tokens (#34332) 2024-10-30 10:21:37 +01:00
rag rag: fix CI (#33578) 2024-09-19 11:55:26 +01:00
recurrent_gemma Make pipeline able to load processor (#32514) 2024-10-09 16:46:11 +01: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 SPLIT PR: add user defined symbols and control symbols (#31305) 2024-06-21 01:48:10 -07: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 Update kwargs validation for preprocess with decorator (#32024) 2024-08-06 11:33:05 +01:00
rwkv Generation tests: update imagegpt input name, remove unused functions (#33663) 2024-09-24 16:40:48 +01:00
sam Make pipeline able to load processor (#32514) 2024-10-09 16:46:11 +01:00
seamless_m4t Generate tests: modality-agnostic input preparation (#33685) 2024-10-03 14:01:24 +01:00
seamless_m4t_v2 Generate tests: modality-agnostic input preparation (#33685) 2024-10-03 14:01:24 +01:00
segformer Skip tests properly (#31308) 2024-06-26 21:59:08 +01:00
seggpt Fast image processor (#28847) 2024-06-11 15:47:38 +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 Attn implementation for composite models (#32238) 2024-10-22 06:54:44 +02:00
speech_encoder_decoder Correct the new defaults (#34377) 2024-10-24 18:42:03 +02:00
speech_to_text Generation tests: don't rely on main input name (#34228) 2024-10-21 10:00:14 +02:00
speecht5 Generate tests: modality-agnostic input preparation (#33685) 2024-10-03 14:01:24 +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 Tests: upgrade test_eager_matches_sdpa_generate (#34386) 2024-10-25 11:55:07 +01:00
starcoder2 Make pipeline able to load processor (#32514) 2024-10-09 16:46:11 +01:00
superpoint 🚨🚨🚨 [SuperPoint] Fix keypoint coordinate output and add post processing (#33200) 2024-10-29 09:36:03 +00:00
swiftformer Rename test_model_common_attributes -> test_model_get_set_embeddings (#31321) 2024-06-07 19:40:26 +01:00
swin Use self.config_tester.run_common_tests() (#31431) 2024-06-19 10:18:08 +01:00
swin2sr Update kwargs validation for preprocess with decorator (#32024) 2024-08-06 11:33:05 +01:00
swinv2 Fix dtype casting in swinv2 and swinv2sr to allow non-FP32 inference (#31589) 2024-06-26 18:46:48 +01:00
switch_transformers T5 compile compatibilty (#34089) 2024-10-22 08:23:53 +02:00
t5 T5 compile compatibilty (#34089) 2024-10-22 08:23:53 +02:00
table_transformer Skip tests properly (#31308) 2024-06-26 21:59:08 +01:00
tapas Make pipeline able to load processor (#32514) 2024-10-09 16:46:11 +01:00
time_series_transformer Skip tests properly (#31308) 2024-06-26 21:59:08 +01:00
timesformer Skip tests properly (#31308) 2024-06-26 21:59:08 +01:00
timm_backbone Skip tests properly (#31308) 2024-06-26 21:59:08 +01:00
trocr Skip tests properly (#31308) 2024-06-26 21:59:08 +01:00
tvp Update kwargs validation for preprocess with decorator (#32024) 2024-08-06 11:33:05 +01:00
udop T5 compile compatibilty (#34089) 2024-10-22 08:23:53 +02:00
umt5 T5 compile compatibilty (#34089) 2024-10-22 08:23:53 +02: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 Generate tests: modality-agnostic input preparation (#33685) 2024-10-03 14:01:24 +01:00
upernet Use self.config_tester.run_common_tests() (#31431) 2024-06-19 10:18:08 +01:00
video_llava VLMs: fix number of image tokens (#34332) 2024-10-30 10:21:37 +01:00
videomae Update kwargs validation for preprocess with decorator (#32024) 2024-08-06 11:33:05 +01:00
vilt Update kwargs validation for preprocess with decorator (#32024) 2024-08-06 11:33:05 +01:00
vipllava VLMs: fix number of image tokens (#34332) 2024-10-30 10:21:37 +01:00
vision_encoder_decoder Correct the new defaults (#34377) 2024-10-24 18:42:03 +02:00
vision_text_dual_encoder tests: fix pytorch tensor placement errors (#33485) 2024-09-25 12:21:53 +01:00
visual_bert update ruff version (#30932) 2024-05-22 06:40:15 +02:00
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 Rename test_model_common_attributes -> test_model_get_set_embeddings (#31321) 2024-06-07 19:40:26 +01:00
vitdet Skip tests properly (#31308) 2024-06-26 21:59:08 +01:00
vitmatte Update kwargs validation for preprocess with decorator (#32024) 2024-08-06 11:33:05 +01: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 Uniformize model processors (#31368) 2024-10-02 10:41:08 +02: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 Pass datasets trust_remote_code (#31406) 2024-06-17 17:29:13 +01:00
wavlm Remove trust_remote_code when loading Libri Dummy (#31748) 2024-07-23 14:54:38 +08:00
whisper Make pipeline able to load processor (#32514) 2024-10-09 16:46:11 +01:00
x_clip adding positional encoder changes and tests (#32600) 2024-09-25 19:05:01 +01:00
xglm Generation tests: update imagegpt input name, remove unused functions (#33663) 2024-09-24 16:40:48 +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 Add DetrImageProcessorFast (#34063) 2024-10-21 09:05:05 -04:00
yoso Skip tests properly (#31308) 2024-06-26 21:59:08 +01:00
zamba Add Zamba (#30950) 2024-10-04 22:28:05 +02: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 Move test model folders (#17034) 2022-05-03 14:42:02 +02:00