riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-31 02:02:21 +06:00
Code Issues Actions 23 Packages Projects Releases Wiki Activity
8e64ba2890
transformers/tests/models
History
Raushan Turganbay 8e64ba2890
Add tests for batching support (#29297) 
* add tests for batching support

* Update src/transformers/models/fastspeech2_conformer/modeling_fastspeech2_conformer.py

Co-authored-by: Joao Gante <joaofranciscocardosogante@gmail.com>

* Update src/transformers/models/fastspeech2_conformer/modeling_fastspeech2_conformer.py

Co-authored-by: Joao Gante <joaofranciscocardosogante@gmail.com>

* Update tests/test_modeling_common.py

Co-authored-by: Joao Gante <joaofranciscocardosogante@gmail.com>

* Update tests/test_modeling_common.py

Co-authored-by: Joao Gante <joaofranciscocardosogante@gmail.com>

* Update tests/test_modeling_common.py

Co-authored-by: Joao Gante <joaofranciscocardosogante@gmail.com>

* fixes and comments

* use cosine distance for conv models

* skip mra model testing

* Update tests/models/vilt/test_modeling_vilt.py

Co-authored-by: Joao Gante <joaofranciscocardosogante@gmail.com>

* finzalize  and make style

* check model type by input names

* Update tests/models/vilt/test_modeling_vilt.py

Co-authored-by: amyeroberts <22614925+amyeroberts@users.noreply.github.com>

* fixed batch size for all testers

* Revert "fixed batch size for all testers"

This reverts commit 525f3a0a05.

* add batch_size for all testers

* dict from model output

* do not skip layoutlm

* bring back some code from git revert

* Update tests/test_modeling_common.py

Co-authored-by: amyeroberts <22614925+amyeroberts@users.noreply.github.com>

* Update tests/test_modeling_common.py

Co-authored-by: amyeroberts <22614925+amyeroberts@users.noreply.github.com>

* clean-up

* where did minus go in tolerance

* make whisper happy

* deal with consequences of losing minus

* deal with consequences of losing minus

* maskformer needs its own test for happiness

* fix more models

* tag flaky CV models from Amy's approval

* make codestyle

---------

Co-authored-by: Joao Gante <joaofranciscocardosogante@gmail.com>
Co-authored-by: amyeroberts <22614925+amyeroberts@users.noreply.github.com>
2024-03-12 17:46:19 +00:00
..
albert Update all references to canonical models (#29001) 2024-02-16 08:16:58 +01:00
align Add tests for batching support (#29297) 2024-03-12 17:46:19 +00:00
altclip Add tests for batching support (#29297) 2024-03-12 17:46:19 +00:00
audio_spectrogram_transformer Add numpy alternative to FE using torchaudio (#26339) 2023-11-08 07:39:37 +00:00
auto Allow remote code repo names to contain "." (#29175) 2024-02-23 12:46:31 +00:00
autoformer Add tests for batching support (#29297) 2024-03-12 17:46:19 +00:00
bark Add tests for batching support (#29297) 2024-03-12 17:46:19 +00:00
bart Generate: left-padding test, revisited (#29515) 2024-03-08 10:06:46 +00:00
barthez [Styling] stylify using ruff (#27144) 2023-11-16 17:43:19 +01:00
bartpho Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
beit Avoid using uncessary get_values(MODEL_MAPPING) (#29362) 2024-02-29 17:19:17 +08:00
bert Update all references to canonical models (#29001) 2024-02-16 08:16:58 +01:00
bert_generation Revert low cpu mem tie weights (#29135) 2024-02-20 12:06:46 +00:00
bert_japanese Update all references to canonical models (#29001) 2024-02-16 08:16:58 +01:00
bertweet Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
big_bird [Styling] stylify using ruff (#27144) 2023-11-16 17:43:19 +01:00
bigbird_pegasus Generate: left-padding test, revisited (#29515) 2024-03-08 10:06:46 +00:00
biogpt [tests] add the missing require_sacremoses decorator (#29504) 2024-03-08 10:13:54 +00:00
bit Image Feature Extraction pipeline (#28216) 2024-02-05 14:50:07 +00:00
blenderbot Generate: left-padding test, revisited (#29515) 2024-03-08 10:06:46 +00:00
blenderbot_small Generate: left-padding test, revisited (#29515) 2024-03-08 10:06:46 +00:00
blip Add tests for batching support (#29297) 2024-03-12 17:46:19 +00:00
blip_2 Add tests for batching support (#29297) 2024-03-12 17:46:19 +00:00
bloom [Styling] stylify using ruff (#27144) 2023-11-16 17:43:19 +01:00
bridgetower Refactor image processor testers (#25450) 2023-08-11 11:30:18 +01:00
bros Update tiny model information and pipeline tests (#26285) 2023-09-25 18:08:12 +02:00
byt5 [Docs] Fix spelling and grammar mistakes (#28825) 2024-02-02 08:45:00 +01:00
camembert Update all references to canonical models (#29001) 2024-02-16 08:16:58 +01:00
canine [Docs] Fix spelling and grammar mistakes (#28825) 2024-02-02 08:45:00 +01:00
chinese_clip Add tests for batching support (#29297) 2024-03-12 17:46:19 +00:00
clap Add tests for batching support (#29297) 2024-03-12 17:46:19 +00:00
clip Add tests for batching support (#29297) 2024-03-12 17:46:19 +00:00
clipseg Add tests for batching support (#29297) 2024-03-12 17:46:19 +00:00
clvp Add tests for batching support (#29297) 2024-03-12 17:46:19 +00:00
code_llama [Docs] Fix spelling and grammar mistakes (#28825) 2024-02-02 08:45:00 +01:00
codegen Device agnostic testing (#25870) 2023-10-24 16:49:26 +02:00
conditional_detr Add tests for batching support (#29297) 2024-03-12 17:46:19 +00:00
convbert Add tf_keras imports to prepare for Keras 3 (#28588) 2024-01-30 17:26:36 +00:00
convnext Image Feature Extraction pipeline (#28216) 2024-02-05 14:50:07 +00:00
convnextv2 Image Feature Extraction pipeline (#28216) 2024-02-05 14:50:07 +00:00
cpm Fix PipelineTests skip conditions (#22320) 2023-03-22 20:02:24 +01:00
cpmant Add tests for batching support (#29297) 2024-03-12 17:46:19 +00:00
ctrl Generate: left-padding test, revisited (#29515) 2024-03-08 10:06:46 +00:00
cvt Image Feature Extraction pipeline (#28216) 2024-02-05 14:50:07 +00:00
data2vec Avoid using uncessary get_values(MODEL_MAPPING) (#29362) 2024-02-29 17:19:17 +08:00
deberta CI with num_hidden_layers=2 🚀🚀🚀 (#25266) 2023-08-02 20:22:36 +02:00
deberta_v2 [Styling] stylify using ruff (#27144) 2023-11-16 17:43:19 +01:00
decision_transformer 🔥Rework pipeline testing by removing PipelineTestCaseMeta 🚀 (#21516) 2023-02-28 19:40:57 +01:00
deformable_detr DeformableDETR support bfloat16 (#29232) 2024-03-04 14:18:09 +00:00
deit Avoid using uncessary get_values(MODEL_MAPPING) (#29362) 2024-02-29 17:19:17 +08:00
depth_anything Add Depth Anything (#28654) 2024-01-25 09:34:50 +01:00
deta [YOLOS] Fix - return padded annotations (#29300) 2024-03-01 09:42:13 +00:00
detr Add tests for batching support (#29297) 2024-03-12 17:46:19 +00:00
dinat Image Feature Extraction pipeline (#28216) 2024-02-05 14:50:07 +00:00
dinov2 Image Feature Extraction pipeline (#28216) 2024-02-05 14:50:07 +00:00
distilbert F.scaled_dot_product_attention support (#26572) 2023-12-09 05:38:14 +09:00
dit Update old existing feature extractor references (#24552) 2023-06-29 10:17:36 +01:00
donut Image Feature Extraction pipeline (#28216) 2024-02-05 14:50:07 +00:00
dpr Update all references to canonical models (#29001) 2024-02-16 08:16:58 +01:00
dpt Add tests for batching support (#29297) 2024-03-12 17:46:19 +00:00
efficientformer Avoid using uncessary get_values(MODEL_MAPPING) (#29362) 2024-02-29 17:19:17 +08:00
efficientnet Image Feature Extraction pipeline (#28216) 2024-02-05 14:50:07 +00:00
electra CI with num_hidden_layers=2 🚀🚀🚀 (#25266) 2023-08-02 20:22:36 +02:00
encodec Add tests for batching support (#29297) 2024-03-12 17:46:19 +00:00
encoder_decoder Generate: missing generation config eos token setting in encoder-decoder tests (#29146) 2024-02-20 16:17:51 +00:00
ernie device agnostic models testing (#27146) 2023-10-31 18:12:14 +01:00
ernie_m [Styling] stylify using ruff (#27144) 2023-11-16 17:43:19 +01:00
esm FIX [quantization / ESM] Fix ESM 8bit / 4bit with bitsandbytes (#29329) 2024-03-01 03:01:53 +01:00
falcon F.scaled_dot_product_attention support (#26572) 2023-12-09 05:38:14 +09:00
fastspeech2_conformer Add tests for batching support (#29297) 2024-03-12 17:46:19 +00:00
flaubert [tests] add the missing require_sacremoses decorator (#29504) 2024-03-08 10:13:54 +00:00
flava Add tests for batching support (#29297) 2024-03-12 17:46:19 +00:00
fnet [Styling] stylify using ruff (#27144) 2023-11-16 17:43:19 +01:00
focalnet Image Feature Extraction pipeline (#28216) 2024-02-05 14:50:07 +00:00
fsmt Revert low cpu mem tie weights (#29135) 2024-02-20 12:06:46 +00:00
funnel Big TF test cleanup (#24282) 2023-06-16 15:40:49 +01:00
fuyu Update tiny model summary file (#27388) 2023-11-23 21:00:39 +01:00
gemma Fix llama + gemma accelete tests (#29380) 2024-03-01 10:32:36 -05:00
git [core/ GC / tests] Stronger GC tests (#27124) 2023-10-30 19:53:46 +01:00
glpn Avoid using uncessary get_values(MODEL_MAPPING) (#29362) 2024-02-29 17:19:17 +08:00
gpt_bigcode CI with num_hidden_layers=2 🚀🚀🚀 (#25266) 2023-08-02 20:22:36 +02:00
gpt_neo Update all references to canonical models (#29001) 2024-02-16 08:16:58 +01:00
gpt_neox [GPTNeoX] Fix BC issue with 4.36 (#28602) 2024-01-21 17:01:19 +00:00
gpt_neox_japanese [Styling] stylify using ruff (#27144) 2023-11-16 17:43:19 +01:00
gpt_sw3 Remove token_type_ids from model_input_names (like #24788) (#28325) 2024-01-03 19:26:07 +01:00
gpt2 Update all references to canonical models (#29001) 2024-02-16 08:16:58 +01:00
gptj Update all references to canonical models (#29001) 2024-02-16 08:16:58 +01:00
gptsan_japanese [Styling] stylify using ruff (#27144) 2023-11-16 17:43:19 +01:00
graphormer Make more test models smaller (#25005) 2023-07-24 10:08:47 -04:00
groupvit Add tests for batching support (#29297) 2024-03-12 17:46:19 +00:00
herbert [tests] add the missing require_sacremoses decorator (#29504) 2024-03-08 10:13:54 +00:00
hubert Fix failing tests on main due to torch 2.1 (#26607) 2023-10-05 10:27:05 +02:00
ibert Update deprecated torch.range in test_modeling_ibert.py (#27355) 2023-11-08 20:58:36 +01:00
idefics F.scaled_dot_product_attention support (#26572) 2023-12-09 05:38:14 +09:00
imagegpt Image Feature Extraction pipeline (#28216) 2024-02-05 14:50:07 +00:00
informer Add tests for batching support (#29297) 2024-03-12 17:46:19 +00:00
instructblip Add tests for batching support (#29297) 2024-03-12 17:46:19 +00:00
jukebox device agnostic models testing (#27146) 2023-10-31 18:12:14 +01:00
kosmos2 Add tests for batching support (#29297) 2024-03-12 17:46:19 +00:00
layoutlm [core/ GC / tests] Stronger GC tests (#27124) 2023-10-30 19:53:46 +01:00
layoutlmv2 Add tests for batching support (#29297) 2024-03-12 17:46:19 +00:00
layoutlmv3 [Styling] stylify using ruff (#27144) 2023-11-16 17:43:19 +01:00
layoutxlm [Styling] stylify using ruff (#27144) 2023-11-16 17:43:19 +01:00
led device agnostic models testing (#27146) 2023-10-31 18:12:14 +01:00
levit Avoid using uncessary get_values(MODEL_MAPPING) (#29362) 2024-02-29 17:19:17 +08:00
lilt [core/ GC / tests] Stronger GC tests (#27124) 2023-10-30 19:53:46 +01:00
llama Fix llama + gemma accelete tests (#29380) 2024-03-01 10:32:36 -05:00
llava Fix _merge_input_ids_with_image_features for llava model (#28333) 2024-01-10 08:33:33 +01:00
longformer Add tests for batching support (#29297) 2024-03-12 17:46:19 +00:00
longt5 Byebye torch 1.10 (#28207) 2024-01-11 16:18:27 +01:00
luke [core/ GC / tests] Stronger GC tests (#27124) 2023-10-30 19:53:46 +01:00
lxmert [Styling] stylify using ruff (#27144) 2023-11-16 17:43:19 +01:00
m2m_100 [Styling] stylify using ruff (#27144) 2023-11-16 17:43:19 +01:00
mamba [Mamba doc] Post merge updates (#29472) 2024-03-11 09:46:24 +01:00
marian Generate: left-padding test, revisited (#29515) 2024-03-08 10:06:46 +00:00
markuplm Update all references to canonical models (#29001) 2024-02-16 08:16:58 +01:00
mask2former Image Feature Extraction pipeline (#28216) 2024-02-05 14:50:07 +00:00
maskformer Add tests for batching support (#29297) 2024-03-12 17:46:19 +00:00
mbart Generate: left-padding test, revisited (#29515) 2024-03-08 10:06:46 +00:00
mbart50 [Styling] stylify using ruff (#27144) 2023-11-16 17:43:19 +01:00
mega fix / skip (for now) some tests before switch to torch 2.2 (#28838) 2024-02-02 14:11:50 +01:00
megatron_bert CI with num_hidden_layers=2 🚀🚀🚀 (#25266) 2023-08-02 20:22:36 +02:00
megatron_gpt2 Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
mgp_str Image Feature Extraction pipeline (#28216) 2024-02-05 14:50:07 +00:00
mistral Flax mistral (#26943) 2024-01-31 14:19:02 +01:00
mixtral Exclude the load balancing loss of padding tokens in Mixtral-8x7B (#28517) 2024-01-24 10:12:14 +01:00
mluke 🚨🚨 🚨🚨 [Tokenizer] attemp to fix add_token issues🚨🚨 🚨🚨 (#23909) 2023-09-18 20:28:36 +02:00
mobilebert Update all references to canonical models (#29001) 2024-02-16 08:16:58 +01:00
mobilenet_v1 Image Feature Extraction pipeline (#28216) 2024-02-05 14:50:07 +00:00
mobilenet_v2 Add tests for batching support (#29297) 2024-03-12 17:46:19 +00:00
mobilevit Image Feature Extraction pipeline (#28216) 2024-02-05 14:50:07 +00:00
mobilevitv2 Image Feature Extraction pipeline (#28216) 2024-02-05 14:50:07 +00:00
mpnet Fix TF loading PT safetensors when weights are tied (#27490) 2023-12-07 14:28:53 +00:00
mpt Fix building alibi tensor when num_heads is not a power of 2 (#28380) 2024-01-08 10:39:40 +01:00
mra Add tests for batching support (#29297) 2024-03-12 17:46:19 +00:00
mt5 Update all references to canonical models (#29001) 2024-02-16 08:16:58 +01:00
musicgen Add tests for batching support (#29297) 2024-03-12 17:46:19 +00:00
mvp Generate: left-padding test, revisited (#29515) 2024-03-08 10:06:46 +00:00
nat Image Feature Extraction pipeline (#28216) 2024-02-05 14:50:07 +00:00
nezha CI with num_hidden_layers=2 🚀🚀🚀 (#25266) 2023-08-02 20:22:36 +02:00
nllb [NllbTokenizer] refactor with added tokens decoder (#27717) 2024-02-13 03:49:20 +01:00
nllb_moe fix assisted decoding assistant model inputs (#27503) 2023-11-27 14:23:54 +00:00
nougat Add Nougat (#25942) 2023-09-26 07:06:04 +02:00
nystromformer CI with num_hidden_layers=2 🚀🚀🚀 (#25266) 2023-08-02 20:22:36 +02:00
oneformer Fix OneFormer post_process_instance_segmentation for panoptic tasks (#29304) 2024-03-04 11:04:49 +00:00
openai Update all references to canonical models (#29001) 2024-02-16 08:16:58 +01:00
opt Proper build() methods for TF (#27794) 2023-12-14 15:17:30 +00:00
owlv2 Add tests for batching support (#29297) 2024-03-12 17:46:19 +00:00
owlvit Add tests for batching support (#29297) 2024-03-12 17:46:19 +00:00
patchtsmixer PatchtTST and PatchTSMixer fixes (#28083) 2024-01-29 10:09:26 +00:00
patchtst PatchtTST and PatchTSMixer fixes (#28083) 2024-01-29 10:09:26 +00:00
pegasus Generate: left-padding test, revisited (#29515) 2024-03-08 10:06:46 +00:00
pegasus_x device agnostic models testing (#27146) 2023-10-31 18:12:14 +01:00
perceiver Avoid using uncessary get_values(MODEL_MAPPING) (#29362) 2024-02-29 17:19:17 +08:00
persimmon Fix device of masks in tests (#27887) 2023-12-07 21:34:43 +09:00
phi [Phi] Extend implementation to use GQA/MQA. (#28163) 2024-01-11 15:58:02 +01:00
phobert Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
pix2struct Add tests for batching support (#29297) 2024-03-12 17:46:19 +00:00
plbart Generate: left-padding test, revisited (#29515) 2024-03-08 10:06:46 +00:00
poolformer Image Feature Extraction pipeline (#28216) 2024-02-05 14:50:07 +00:00
pop2piano Byebye torch 1.10 (#28207) 2024-01-11 16:18:27 +01:00
prophetnet Generate: left-padding test, revisited (#29515) 2024-03-08 10:06:46 +00:00
pvt Avoid using uncessary get_values(MODEL_MAPPING) (#29362) 2024-02-29 17:19:17 +08:00
qdqbert Update all references to canonical models (#29001) 2024-02-16 08:16:58 +01:00
qwen2 [Docs] Fix spelling and grammar mistakes (#28825) 2024-02-02 08:45:00 +01:00
rag Even more TF test fixes (#28146) 2023-12-21 15:14:46 +00:00
realm Update all references to canonical models (#29001) 2024-02-16 08:16:58 +01:00
reformer Revert low cpu mem tie weights (#29135) 2024-02-20 12:06:46 +00:00
regnet Image Feature Extraction pipeline (#28216) 2024-02-05 14:50:07 +00:00
rembert Added test cases for rembert refering to albert and reformer test_tok… (#27637) 2023-12-04 13:36:57 +01:00
resnet Image Feature Extraction pipeline (#28216) 2024-02-05 14:50:07 +00:00
roberta Update all references to canonical models (#29001) 2024-02-16 08:16:58 +01:00
roberta_prelayernorm Update all references to canonical models (#29001) 2024-02-16 08:16:58 +01:00
roc_bert Copied from for test files (#26713) 2023-10-11 14:12:09 +02:00
roformer [TokenizationRoformerFast] Fix the save and loading (#28527) 2024-01-16 16:37:15 +01:00
rwkv CI with num_hidden_layers=2 🚀🚀🚀 (#25266) 2023-08-02 20:22:36 +02:00
sam Add tf_keras imports to prepare for Keras 3 (#28588) 2024-01-30 17:26:36 +00:00
seamless_m4t [Core tokenization] add_dummy_prefix_space option to help with latest issues (#28010) 2024-02-20 12:50:31 +01:00
seamless_m4t_v2 Fix languages covered by M4Tv2 (#28019) 2023-12-14 14:43:44 +00:00
segformer Avoid using uncessary get_values(MODEL_MAPPING) (#29362) 2024-02-29 17:19:17 +08:00
seggpt Adding SegGPT (#27735) 2024-02-26 18:17:19 +00:00
sew Fix typo (#25966) 2023-09-05 10:12:25 +02:00
sew_d Fix typo (#25966) 2023-09-05 10:12:25 +02:00
siglip Add tests for batching support (#29297) 2024-03-12 17:46:19 +00:00
speech_encoder_decoder Generate: missing generation config eos token setting in encoder-decoder tests (#29146) 2024-02-20 16:17:51 +00:00
speech_to_text [Styling] stylify using ruff (#27144) 2023-11-16 17:43:19 +01:00
speech_to_text_2 Fix TF loading PT safetensors when weights are tied (#27490) 2023-12-07 14:28:53 +00:00
speecht5 Add tests for batching support (#29297) 2024-03-12 17:46:19 +00:00
splinter CI with num_hidden_layers=2 🚀🚀🚀 (#25266) 2023-08-02 20:22:36 +02:00
squeezebert CI with num_hidden_layers=2 🚀🚀🚀 (#25266) 2023-08-02 20:22:36 +02:00
stablelm support SDPA Attention in stablelm (#29106) 2024-02-21 13:12:49 +01:00
starcoder2 FIX [CI / starcoder2] Change starcoder2 path to correct one for slow tests (#29359) 2024-02-29 03:52:13 +01:00
swiftformer Image Feature Extraction pipeline (#28216) 2024-02-05 14:50:07 +00:00
swin Image Feature Extraction pipeline (#28216) 2024-02-05 14:50:07 +00:00
swin2sr Image Feature Extraction pipeline (#28216) 2024-02-05 14:50:07 +00:00
swinv2 Image Feature Extraction pipeline (#28216) 2024-02-05 14:50:07 +00:00
switch_transformers Update all references to canonical models (#29001) 2024-02-16 08:16:58 +01:00
t5 [Core tokenization] add_dummy_prefix_space option to help with latest issues (#28010) 2024-02-20 12:50:31 +01:00
table_transformer Add tests for batching support (#29297) 2024-03-12 17:46:19 +00:00
tapas [Styling] stylify using ruff (#27144) 2023-11-16 17:43:19 +01:00
time_series_transformer Add tests for batching support (#29297) 2024-03-12 17:46:19 +00:00
timesformer Update forward signature test for vision models (#27681) 2023-11-27 15:48:17 +01:00
timm_backbone [AutoBackbone] Add test (#26094) 2023-09-18 23:47:54 +02:00
trocr CI with num_hidden_layers=2 🚀🚀🚀 (#25266) 2023-08-02 20:22:36 +02:00
tvlt remove failing tests and clean FE files (#27414) 2023-11-09 18:35:42 +00:00
tvp Enable instantiating model with pretrained backbone weights (#28214) 2024-01-23 11:01:50 +00:00
udop [Udop imports] Processor tests were not run. (#29456) 2024-03-05 11:01:08 +01:00
umt5 Update all references to canonical models (#29001) 2024-02-16 08:16:58 +01:00
unispeech [Wav2Vec2 and Co] Update init tests for PT 2.1 (#26494) 2023-10-03 10:52:34 +02:00
unispeech_sat Byebye torch 1.10 (#28207) 2024-01-11 16:18:27 +01:00
univnet Add tests for batching support (#29297) 2024-03-12 17:46:19 +00:00
upernet Enable instantiating model with pretrained backbone weights (#28214) 2024-01-23 11:01:50 +00:00
videomae Update forward signature test for vision models (#27681) 2023-11-27 15:48:17 +01:00
vilt Add tests for batching support (#29297) 2024-03-12 17:46:19 +00:00
vipllava Fix _merge_input_ids_with_image_features for llava model (#28333) 2024-01-10 08:33:33 +01:00
vision_encoder_decoder Fix VisionEncoderDecoder Positional Arg (#29497) 2024-03-07 20:45:51 +00:00
vision_text_dual_encoder [Styling] stylify using ruff (#27144) 2023-11-16 17:43:19 +01:00
visual_bert [core/ GC / tests] Stronger GC tests (#27124) 2023-10-30 19:53:46 +01:00
vit Image Feature Extraction pipeline (#28216) 2024-02-05 14:50:07 +00:00
vit_hybrid Add tests for batching support (#29297) 2024-03-12 17:46:19 +00:00
vit_mae Add tests for batching support (#29297) 2024-03-12 17:46:19 +00:00
vit_msn Image Feature Extraction pipeline (#28216) 2024-02-05 14:50:07 +00:00
vitdet mark test_initialization as flaky in 2 model tests (#27906) 2023-12-08 14:54:32 +01:00
vitmatte Enable instantiating model with pretrained backbone weights (#28214) 2024-01-23 11:01:50 +00:00
vits Add tests for batching support (#29297) 2024-03-12 17:46:19 +00:00
vivit Input data format (#25464) 2023-08-16 17:45:02 +01:00
wav2vec2 Byebye torch 1.10 (#28207) 2024-01-11 16:18:27 +01:00
wav2vec2_bert Add new meta w2v2-conformer BERT-like model (#28165) 2024-01-18 13:37:34 +00:00
wav2vec2_conformer device agnostic models testing (#27146) 2023-10-31 18:12:14 +01:00
wav2vec2_phoneme Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
wav2vec2_with_lm Fix some tests using "common_voice" (#27147) 2023-10-30 15:27:15 +01:00
wavlm Byebye torch 1.10 (#28207) 2024-01-11 16:18:27 +01:00
whisper Add tests for batching support (#29297) 2024-03-12 17:46:19 +00:00
x_clip Add tests for batching support (#29297) 2024-03-12 17:46:19 +00:00
xglm Fix TF loading PT safetensors when weights are tied (#27490) 2023-12-07 14:28:53 +00:00
xlm Update all references to canonical models (#29001) 2024-02-16 08:16:58 +01:00
xlm_prophetnet [Styling] stylify using ruff (#27144) 2023-11-16 17:43:19 +01:00
xlm_roberta Update all references to canonical models (#29001) 2024-02-16 08:16:58 +01:00
xlm_roberta_xl Revert low cpu mem tie weights (#29135) 2024-02-20 12:06:46 +00:00
xlnet Update all references to canonical models (#29001) 2024-02-16 08:16:58 +01:00
xmod Update all references to canonical models (#29001) 2024-02-16 08:16:58 +01:00
yolos [YOLOS] Fix - return padded annotations (#29300) 2024-03-01 09:42:13 +00:00
yoso CI with num_hidden_layers=2 🚀🚀🚀 (#25266) 2023-08-02 20:22:36 +02:00
__init__.py Move test model folders (#17034) 2022-05-03 14:42:02 +02:00