transformers/tests/models
2024-02-02 16:57:08 +00:00
..
albert [Styling] stylify using ruff (#27144) 2023-11-16 17:43:19 +01:00
align [core/ GC / tests] Stronger GC tests (#27124) 2023-10-30 19:53:46 +01:00
altclip [Styling] stylify using ruff (#27144) 2023-11-16 17:43:19 +01:00
audio_spectrogram_transformer Add numpy alternative to FE using torchaudio (#26339) 2023-11-08 07:39:37 +00:00
auto Don't save processor_config.json if a processor has no extra attribute (#28584) 2024-01-19 09:59:14 +00:00
autoformer [core/ GC / tests] Stronger GC tests (#27124) 2023-10-30 19:53:46 +01:00
bark F.scaled_dot_product_attention support (#26572) 2023-12-09 05:38:14 +09:00
bart Replace build() with build_in_name_scope() for some TF tests (#28046) 2023-12-14 17:42:25 +00:00
barthez [Styling] stylify using ruff (#27144) 2023-11-16 17:43:19 +01:00
bartpho
beit Add BeitBackbone (#25952) 2023-11-28 08:38:32 +00:00
bert Add tf_keras imports to prepare for Keras 3 (#28588) 2024-01-30 17:26:36 +00:00
bert_generation [Styling] stylify using ruff (#27144) 2023-11-16 17:43:19 +01:00
bert_japanese [Styling] stylify using ruff (#27144) 2023-11-16 17:43:19 +01:00
bertweet
big_bird [Styling] stylify using ruff (#27144) 2023-11-16 17:43:19 +01:00
bigbird_pegasus device agnostic models testing (#27146) 2023-10-31 18:12:14 +01:00
biogpt
bit Update forward signature test for vision models (#27681) 2023-11-27 15:48:17 +01:00
blenderbot device agnostic models testing (#27146) 2023-10-31 18:12:14 +01:00
blenderbot_small device agnostic models testing (#27146) 2023-10-31 18:12:14 +01:00
blip Add tf_keras imports to prepare for Keras 3 (#28588) 2024-01-30 17:26:36 +00:00
blip_2 device agnostic models testing (#27146) 2023-10-31 18:12:14 +01:00
bloom [Styling] stylify using ruff (#27144) 2023-11-16 17:43:19 +01:00
bridgetower
bros
byt5 [Docs] Fix spelling and grammar mistakes (#28825) 2024-02-02 08:45:00 +01:00
camembert [Styling] stylify using ruff (#27144) 2023-11-16 17:43:19 +01:00
canine [Docs] Fix spelling and grammar mistakes (#28825) 2024-02-02 08:45:00 +01:00
chinese_clip [Styling] stylify using ruff (#27144) 2023-11-16 17:43:19 +01:00
clap [core/ GC / tests] Stronger GC tests (#27124) 2023-10-30 19:53:46 +01:00
clip Add tf_keras imports to prepare for Keras 3 (#28588) 2024-01-30 17:26:36 +00:00
clipseg [Styling] stylify using ruff (#27144) 2023-11-16 17:43:19 +01:00
clvp CLVP Fixes (#27547) 2023-11-28 17:40:01 +01: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 [Backbone] Use load_backbone instead of AutoBackbone.from_config (#28661) 2024-01-30 16:54:09 +00:00
convbert Add tf_keras imports to prepare for Keras 3 (#28588) 2024-01-30 17:26:36 +00:00
convnext Update forward signature test for vision models (#27681) 2023-11-27 15:48:17 +01:00
convnextv2 Update forward signature test for vision models (#27681) 2023-11-27 15:48:17 +01:00
cpm
cpmant
ctrl Add tf_keras imports to prepare for Keras 3 (#28588) 2024-01-30 17:26:36 +00:00
cvt Add tf_keras imports to prepare for Keras 3 (#28588) 2024-01-30 17:26:36 +00:00
data2vec Add tf_keras imports to prepare for Keras 3 (#28588) 2024-01-30 17:26:36 +00:00
deberta
deberta_v2 [Styling] stylify using ruff (#27144) 2023-11-16 17:43:19 +01:00
decision_transformer
deformable_detr Enable instantiating model with pretrained backbone weights (#28214) 2024-01-23 11:01:50 +00:00
deit Add tf_keras imports to prepare for Keras 3 (#28588) 2024-01-30 17:26:36 +00:00
depth_anything Add Depth Anything (#28654) 2024-01-25 09:34:50 +01:00
deta Enable instantiating model with pretrained backbone weights (#28214) 2024-01-23 11:01:50 +00:00
detr Enable instantiating model with pretrained backbone weights (#28214) 2024-01-23 11:01:50 +00:00
dinat Update forward signature test for vision models (#27681) 2023-11-27 15:48:17 +01:00
dinov2 mark test_initialization as flaky in 2 model tests (#27906) 2023-12-08 14:54:32 +01:00
distilbert F.scaled_dot_product_attention support (#26572) 2023-12-09 05:38:14 +09:00
dit
donut Update forward signature test for vision models (#27681) 2023-11-27 15:48:17 +01:00
dpr
dpt Enable instantiating model with pretrained backbone weights (#28214) 2024-01-23 11:01:50 +00:00
efficientformer Add tf_keras imports to prepare for Keras 3 (#28588) 2024-01-30 17:26:36 +00:00
efficientnet Update forward signature test for vision models (#27681) 2023-11-27 15:48:17 +01:00
electra
encodec
encoder_decoder Add tf_keras imports to prepare for Keras 3 (#28588) 2024-01-30 17:26:36 +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 Add tf_keras imports to prepare for Keras 3 (#28588) 2024-01-30 17:26:36 +00:00
falcon F.scaled_dot_product_attention support (#26572) 2023-12-09 05:38:14 +09:00
fastspeech2_conformer Add FastSpeech2Conformer (#23439) 2024-01-03 18:01:06 +00:00
flaubert device agnostic models testing (#27146) 2023-10-31 18:12:14 +01:00
flava Fix pos_mask application and update tests accordingly (#27892) 2024-01-05 12:36:10 +01:00
fnet [Styling] stylify using ruff (#27144) 2023-11-16 17:43:19 +01:00
focalnet Update forward signature test for vision models (#27681) 2023-11-27 15:48:17 +01:00
fsmt device agnostic models testing (#27146) 2023-10-31 18:12:14 +01:00
funnel
fuyu Update tiny model summary file (#27388) 2023-11-23 21:00:39 +01:00
git [core/ GC / tests] Stronger GC tests (#27124) 2023-10-30 19:53:46 +01:00
glpn Update forward signature test for vision models (#27681) 2023-11-27 15:48:17 +01:00
gpt2 Add tf_keras imports to prepare for Keras 3 (#28588) 2024-01-30 17:26:36 +00:00
gpt_bigcode
gpt_neo [Styling] stylify using ruff (#27144) 2023-11-16 17:43:19 +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
gptj [Styling] stylify using ruff (#27144) 2023-11-16 17:43:19 +01:00
gptsan_japanese [Styling] stylify using ruff (#27144) 2023-11-16 17:43:19 +01:00
graphormer
groupvit Add tf_keras imports to prepare for Keras 3 (#28588) 2024-01-30 17:26:36 +00:00
herbert [Tokenizer] Fix slow and fast serialization (#26570) 2023-10-18 16:30:53 +02:00
hubert
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 Update fixtures-image-utils (#28080) 2023-12-15 16:58:36 +00:00
informer [core/ GC / tests] Stronger GC tests (#27124) 2023-10-30 19:53:46 +01:00
instructblip [Styling] stylify using ruff (#27144) 2023-11-16 17:43:19 +01:00
jukebox device agnostic models testing (#27146) 2023-10-31 18:12:14 +01:00
kosmos2 Make image processors more general (#27690) 2023-12-05 10:45:39 +01:00
layoutlm [core/ GC / tests] Stronger GC tests (#27124) 2023-10-30 19:53:46 +01:00
layoutlmv2 [Styling] stylify using ruff (#27144) 2023-11-16 17:43:19 +01: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 Update forward signature test for vision models (#27681) 2023-11-27 15:48:17 +01:00
lilt [core/ GC / tests] Stronger GC tests (#27124) 2023-10-30 19:53:46 +01:00
llama [Docs] Fix spelling and grammar mistakes (#28825) 2024-02-02 08:45:00 +01:00
llava Fix _merge_input_ids_with_image_features for llava model (#28333) 2024-01-10 08:33:33 +01:00
longformer Allow # Ignore copy (#27328) 2023-12-07 10:00:08 +01: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
marian [Styling] stylify using ruff (#27144) 2023-11-16 17:43:19 +01:00
markuplm [Styling] stylify using ruff (#27144) 2023-11-16 17:43:19 +01:00
mask2former Enable instantiating model with pretrained backbone weights (#28214) 2024-01-23 11:01:50 +00:00
maskformer Enable instantiating model with pretrained backbone weights (#28214) 2024-01-23 11:01:50 +00:00
mbart device agnostic models testing (#27146) 2023-10-31 18:12:14 +01: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
megatron_gpt2
mgp_str Update forward signature test for vision models (#27681) 2023-11-27 15:48:17 +01: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
mobilebert Fix TF loading PT safetensors when weights are tied (#27490) 2023-12-07 14:28:53 +00:00
mobilenet_v1 Update forward signature test for vision models (#27681) 2023-11-27 15:48:17 +01:00
mobilenet_v2 Update forward signature test for vision models (#27681) 2023-11-27 15:48:17 +01:00
mobilevit Optionally preprocess segmentation maps for MobileViT (#28420) 2024-01-11 14:52:14 +00:00
mobilevitv2 Update forward signature test for vision models (#27681) 2023-11-27 15:48:17 +01: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 [core/ GC / tests] Stronger GC tests (#27124) 2023-10-30 19:53:46 +01:00
mt5 Adding [T5/MT5/UMT5]ForTokenClassification (#28443) 2024-02-01 03:53:49 +01:00
musicgen Generate: consolidate output classes (#28494) 2024-01-15 17:04:08 +00:00
mvp [Styling] stylify using ruff (#27144) 2023-11-16 17:43:19 +01:00
nat Update forward signature test for vision models (#27681) 2023-11-27 15:48:17 +01:00
nezha
nllb [Styling] stylify using ruff (#27144) 2023-11-16 17:43:19 +01:00
nllb_moe fix assisted decoding assistant model inputs (#27503) 2023-11-27 14:23:54 +00:00
nougat
nystromformer
oneformer Enable instantiating model with pretrained backbone weights (#28214) 2024-01-23 11:01:50 +00:00
openai
opt Proper build() methods for TF (#27794) 2023-12-14 15:17:30 +00:00
owlv2 Fix Owlv2ModelIntegrationTest::test_inference_object_detection (#27793) 2023-12-04 09:45:22 +01:00
owlvit [Styling] stylify using ruff (#27144) 2023-11-16 17:43:19 +01: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 [Styling] stylify using ruff (#27144) 2023-11-16 17:43:19 +01:00
pegasus_x device agnostic models testing (#27146) 2023-10-31 18:12:14 +01:00
perceiver [Docs] Fix spelling and grammar mistakes (#28825) 2024-02-02 08:45:00 +01: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
pix2struct Byebye torch 1.10 (#28207) 2024-01-11 16:18:27 +01:00
plbart device agnostic models testing (#27146) 2023-10-31 18:12:14 +01:00
poolformer Update forward signature test for vision models (#27681) 2023-11-27 15:48:17 +01:00
pop2piano Byebye torch 1.10 (#28207) 2024-01-11 16:18:27 +01:00
prophetnet
pvt Update forward signature test for vision models (#27681) 2023-11-27 15:48:17 +01:00
qdqbert
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
reformer [Styling] stylify using ruff (#27144) 2023-11-16 17:43:19 +01:00
regnet Update forward signature test for vision models (#27681) 2023-11-27 15:48:17 +01:00
rembert Added test cases for rembert refering to albert and reformer test_tok… (#27637) 2023-12-04 13:36:57 +01:00
resnet Update forward signature test for vision models (#27681) 2023-11-27 15:48:17 +01:00
roberta
roberta_prelayernorm [Styling] stylify using ruff (#27144) 2023-11-16 17:43:19 +01:00
roc_bert
roformer [TokenizationRoformerFast] Fix the save and loading (#28527) 2024-01-16 16:37:15 +01:00
rwkv
sam Add tf_keras imports to prepare for Keras 3 (#28588) 2024-01-30 17:26:36 +00:00
seamless_m4t Fix error in M4T feature extractor (#28340) 2024-01-04 16:40:53 +00:00
seamless_m4t_v2 Fix languages covered by M4Tv2 (#28019) 2023-12-14 14:43:44 +00:00
segformer Update forward signature test for vision models (#27681) 2023-11-27 15:48:17 +01:00
sew
sew_d
siglip Add SigLIP (#26522) 2024-01-08 18:17:16 +01:00
speech_encoder_decoder
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 [SpeechT5Tokenization] Add copied from and fix the convert_tokens_to_string to match the fast decoding scheme (#28522) 2024-01-16 16:50:02 +01:00
splinter
squeezebert
swiftformer Update forward signature test for vision models (#27681) 2023-11-27 15:48:17 +01:00
swin Add tf_keras imports to prepare for Keras 3 (#28588) 2024-01-30 17:26:36 +00:00
swin2sr Update forward signature test for vision models (#27681) 2023-11-27 15:48:17 +01:00
swinv2 Add Swinv2 backbone (#27742) 2023-12-22 11:12:56 +00:00
switch_transformers fix assisted decoding assistant model inputs (#27503) 2023-11-27 14:23:54 +00:00
t5 Adding [T5/MT5/UMT5]ForTokenClassification (#28443) 2024-02-01 03:53:49 +01:00
table_transformer Enable instantiating model with pretrained backbone weights (#28214) 2024-01-23 11:01:50 +00:00
tapas [Styling] stylify using ruff (#27144) 2023-11-16 17:43:19 +01:00
time_series_transformer [core/ GC / tests] Stronger GC tests (#27124) 2023-10-30 19:53:46 +01:00
timesformer Update forward signature test for vision models (#27681) 2023-11-27 15:48:17 +01:00
timm_backbone
trocr
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
umt5 Adding [T5/MT5/UMT5]ForTokenClassification (#28443) 2024-02-01 03:53:49 +01:00
unispeech
unispeech_sat Byebye torch 1.10 (#28207) 2024-01-11 16:18:27 +01:00
univnet Skip UnivNetModelTest::test_multi_gpu_data_parallel_forward (#27912) 2023-12-11 09:17:37 +01: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 [core/ GC / tests] Stronger GC tests (#27124) 2023-10-30 19:53:46 +01:00
vipllava Fix _merge_input_ids_with_image_features for llava model (#28333) 2024-01-10 08:33:33 +01:00
vision_encoder_decoder Mark test_encoder_decoder_model_generate for vision_encoder_deocder as flaky (#28842) 2024-02-02 16:57:08 +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 Add tf_keras imports to prepare for Keras 3 (#28588) 2024-01-30 17:26:36 +00:00
vit_hybrid Enable instantiating model with pretrained backbone weights (#28214) 2024-01-23 11:01:50 +00:00
vit_mae Add tf_keras imports to prepare for Keras 3 (#28588) 2024-01-30 17:26:36 +00:00
vit_msn [CI slow] Fix expected values (#27999) 2023-12-13 13:37:10 +01: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 fix / skip (for now) some tests before switch to torch 2.2 (#28838) 2024-02-02 14:11:50 +01:00
vivit
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
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 [Whisper] Refactor forced_decoder_ids & prompt ids (#28687) 2024-01-31 14:02:07 +02:00
x_clip [core/ GC / tests] Stronger GC tests (#27124) 2023-10-30 19:53:46 +01:00
xglm Fix TF loading PT safetensors when weights are tied (#27490) 2023-12-07 14:28:53 +00:00
xlm
xlm_prophetnet [Styling] stylify using ruff (#27144) 2023-11-16 17:43:19 +01:00
xlm_roberta [Styling] stylify using ruff (#27144) 2023-11-16 17:43:19 +01:00
xlm_roberta_xl
xlnet [Styling] stylify using ruff (#27144) 2023-11-16 17:43:19 +01:00
xmod
yolos Update YOLOS slow test values (#28187) 2023-12-21 18:17:07 +00:00
yoso
__init__.py