transformers/tests/models
Yoach Lacombe d355741eca
Fix pad_to_max_length Whisper (#30787)
* fix pad_to_max_length Whisper

* add tests

* make style
2024-05-27 16:09:05 +02:00
..
albert Remove static pretrained maps from the library's internals (#29112) 2024-03-25 10:33:38 +01:00
align update ruff version (#30932) 2024-05-22 06:40:15 +02:00
altclip update ruff version (#30932) 2024-05-22 06:40:15 +02:00
audio_spectrogram_transformer update ruff version (#30932) 2024-05-22 06:40:15 +02:00
auto Fix auto tests (#30067) 2024-04-05 17:49:46 +02:00
autoformer update ruff version (#30932) 2024-05-22 06:40:15 +02:00
bark update ruff version (#30932) 2024-05-22 06:40:15 +02:00
bart update ruff version (#30932) 2024-05-22 06:40:15 +02:00
barthez Adds pretrained IDs directly in the tests (#29534) 2024-03-13 14:53:27 +01:00
bartpho Adds pretrained IDs directly in the tests (#29534) 2024-03-13 14:53:27 +01:00
beit update ruff version (#30932) 2024-05-22 06:40:15 +02:00
bert Fix accelerate failing tests (#30836) 2024-05-23 17:18:58 +02: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 update ruff version (#30932) 2024-05-22 06:40:15 +02:00
bigbird_pegasus update ruff version (#30932) 2024-05-22 06:40:15 +02:00
biogpt update ruff version (#30932) 2024-05-22 06:40:15 +02:00
bit update ruff version (#30932) 2024-05-22 06:40:15 +02:00
blenderbot update ruff version (#30932) 2024-05-22 06:40:15 +02:00
blenderbot_small update ruff version (#30932) 2024-05-22 06:40:15 +02:00
blip update ruff version (#30932) 2024-05-22 06:40:15 +02:00
blip_2 update ruff version (#30932) 2024-05-22 06:40:15 +02:00
bloom Remove static pretrained maps from the library's internals (#29112) 2024-03-25 10:33:38 +01:00
bridgetower update ruff version (#30932) 2024-05-22 06:40:15 +02:00
bros update ruff version (#30932) 2024-05-22 06:40:15 +02:00
byt5 update ruff version (#30932) 2024-05-22 06:40:15 +02:00
camembert Adds pretrained IDs directly in the tests (#29534) 2024-03-13 14:53:27 +01:00
canine update ruff version (#30932) 2024-05-22 06:40:15 +02:00
chinese_clip update ruff version (#30932) 2024-05-22 06:40:15 +02:00
clap update ruff version (#30932) 2024-05-22 06:40:15 +02:00
clip update ruff version (#30932) 2024-05-22 06:40:15 +02:00
clipseg update ruff version (#30932) 2024-05-22 06:40:15 +02:00
clvp update ruff version (#30932) 2024-05-22 06:40:15 +02:00
code_llama Adds pretrained IDs directly in the tests (#29534) 2024-03-13 14:53:27 +01:00
codegen Add token type ids to CodeGenTokenizer (#29265) 2024-04-17 12:19:18 +02:00
cohere update ruff version (#30932) 2024-05-22 06:40:15 +02:00
conditional_detr update ruff version (#30932) 2024-05-22 06:40:15 +02:00
convbert update ruff version (#30932) 2024-05-22 06:40:15 +02:00
convnext update ruff version (#30932) 2024-05-22 06:40:15 +02:00
convnextv2 update ruff version (#30932) 2024-05-22 06:40:15 +02:00
cpm Fix PipelineTests skip conditions (#22320) 2023-03-22 20:02:24 +01:00
cpmant update ruff version (#30932) 2024-05-22 06:40:15 +02:00
ctrl Remove static pretrained maps from the library's internals (#29112) 2024-03-25 10:33:38 +01:00
cvt update ruff version (#30932) 2024-05-22 06:40:15 +02:00
data2vec update ruff version (#30932) 2024-05-22 06:40:15 +02:00
dbrx Fix accelerate failing tests (#30836) 2024-05-23 17:18:58 +02:00
deberta Remove static pretrained maps from the library's internals (#29112) 2024-03-25 10:33:38 +01:00
deberta_v2 Remove static pretrained maps from the library's internals (#29112) 2024-03-25 10:33:38 +01:00
decision_transformer update ruff version (#30932) 2024-05-22 06:40:15 +02:00
deformable_detr update ruff version (#30932) 2024-05-22 06:40:15 +02:00
deit update ruff version (#30932) 2024-05-22 06:40:15 +02:00
depth_anything update ruff version (#30932) 2024-05-22 06:40:15 +02:00
deta update ruff version (#30932) 2024-05-22 06:40:15 +02:00
detr update ruff version (#30932) 2024-05-22 06:40:15 +02:00
dinat update ruff version (#30932) 2024-05-22 06:40:15 +02:00
dinov2 update ruff version (#30932) 2024-05-22 06:40:15 +02:00
distilbert Fix FA2 tests (#29909) 2024-04-01 07:51:00 +00:00
dit Update old existing feature extractor references (#24552) 2023-06-29 10:17:36 +01:00
donut update ruff version (#30932) 2024-05-22 06:40:15 +02:00
dpr Remove static pretrained maps from the library's internals (#29112) 2024-03-25 10:33:38 +01:00
dpt update ruff version (#30932) 2024-05-22 06:40:15 +02:00
efficientformer update ruff version (#30932) 2024-05-22 06:40:15 +02:00
efficientnet update ruff version (#30932) 2024-05-22 06:40:15 +02:00
electra update ruff version (#30932) 2024-05-22 06:40:15 +02:00
encodec update ruff version (#30932) 2024-05-22 06:40:15 +02:00
encoder_decoder [tests] add `torch.use_deterministic_algorithms` for XPU (#30774) 2024-05-23 16:53:07 +01:00
ernie Remove static pretrained maps from the library's internals (#29112) 2024-03-25 10:33:38 +01:00
ernie_m update ruff version (#30932) 2024-05-22 06:40:15 +02:00
esm update ruff version (#30932) 2024-05-22 06:40:15 +02:00
falcon update ruff version (#30932) 2024-05-22 06:40:15 +02:00
fastspeech2_conformer update ruff version (#30932) 2024-05-22 06:40:15 +02:00
flaubert Fix 29807 sinusoidal positional encodings in Flaubert, Informer and XLM (#29904) 2024-04-02 10:27:26 +02:00
flava update ruff version (#30932) 2024-05-22 06:40:15 +02:00
fnet update ruff version (#30932) 2024-05-22 06:40:15 +02:00
focalnet update ruff version (#30932) 2024-05-22 06:40:15 +02:00
fsmt Encoder-decoder models: move embedding scale to nn.Module (#30410) 2024-05-01 12:33:00 +05:00
funnel update ruff version (#30932) 2024-05-22 06:40:15 +02:00
fuyu update ruff version (#30932) 2024-05-22 06:40:15 +02:00
gemma update ruff version (#30932) 2024-05-22 06:40:15 +02:00
git fix: Replace deprecated `assertEquals` with `assertEqual` (#30241) 2024-04-15 09:36:06 +01:00
glpn update ruff version (#30932) 2024-05-22 06:40:15 +02:00
gpt2 update ruff version (#30932) 2024-05-22 06:40:15 +02:00
gpt_bigcode CI with `num_hidden_layers=2` 🚀🚀🚀 (#25266) 2023-08-02 20:22:36 +02:00
gpt_neo update ruff version (#30932) 2024-05-22 06:40:15 +02:00
gpt_neox update ruff version (#30932) 2024-05-22 06:40:15 +02:00
gpt_neox_japanese update ruff version (#30932) 2024-05-22 06:40:15 +02:00
gpt_sw3 Adds pretrained IDs directly in the tests (#29534) 2024-03-13 14:53:27 +01:00
gptj Remove static pretrained maps from the library's internals (#29112) 2024-03-25 10:33:38 +01:00
gptsan_japanese Fix accelerate failing tests (#30836) 2024-05-23 17:18:58 +02:00
graphormer update ruff version (#30932) 2024-05-22 06:40:15 +02:00
grounding_dino update ruff version (#30932) 2024-05-22 06:40:15 +02:00
groupvit update ruff version (#30932) 2024-05-22 06:40:15 +02:00
herbert Adds pretrained IDs directly in the tests (#29534) 2024-03-13 14:53:27 +01:00
hubert update ruff version (#30932) 2024-05-22 06:40:15 +02:00
ibert Encoder-decoder models: move embedding scale to nn.Module (#30410) 2024-05-01 12:33:00 +05:00
idefics update ruff version (#30932) 2024-05-22 06:40:15 +02:00
idefics2 Encoder-decoder models: move embedding scale to nn.Module (#30410) 2024-05-01 12:33:00 +05:00
imagegpt Encoder-decoder models: move embedding scale to nn.Module (#30410) 2024-05-01 12:33:00 +05:00
informer update ruff version (#30932) 2024-05-22 06:40:15 +02:00
instructblip update ruff version (#30932) 2024-05-22 06:40:15 +02:00
jamba update ruff version (#30932) 2024-05-22 06:40:15 +02:00
jetmoe Add JetMoE model (#30005) 2024-05-14 16:32:01 +02:00
jukebox Tests: replace `torch.testing.assert_allclose` by `torch.testing.assert_close` (#29915) 2024-03-28 09:53:31 +00:00
kosmos2 update ruff version (#30932) 2024-05-22 06:40:15 +02:00
layoutlm update ruff version (#30932) 2024-05-22 06:40:15 +02:00
layoutlmv2 update ruff version (#30932) 2024-05-22 06:40:15 +02:00
layoutlmv3 update ruff version (#30932) 2024-05-22 06:40:15 +02:00
layoutxlm Add split special tokens (#30772) 2024-05-24 08:38:58 -07:00
led update ruff version (#30932) 2024-05-22 06:40:15 +02:00
levit update ruff version (#30932) 2024-05-22 06:40:15 +02:00
lilt Remove static pretrained maps from the library's internals (#29112) 2024-03-25 10:33:38 +01:00
llama add prefix space ignored in llama #29625 (#30964) 2024-05-24 01:03:00 -07:00
llava Support arbitrary processor (#30875) 2024-05-17 16:51:31 +02:00
llava_next LLaVa-Next: Update docs with batched inference (#30857) 2024-05-20 13:45:56 +05:00
longformer update ruff version (#30932) 2024-05-22 06:40:15 +02:00
longt5 Do not remove half seq length in generation tests (#30016) 2024-04-19 17:32:52 +01:00
luke update ruff version (#30932) 2024-05-22 06:40:15 +02:00
lxmert update ruff version (#30932) 2024-05-22 06:40:15 +02:00
m2m_100 update ruff version (#30932) 2024-05-22 06:40:15 +02:00
mamba Mamba `slow_forward` gradient fix (#29563) 2024-03-27 04:52:12 +01:00
marian update ruff version (#30932) 2024-05-22 06:40:15 +02:00
markuplm update ruff version (#30932) 2024-05-22 06:40:15 +02:00
mask2former update ruff version (#30932) 2024-05-22 06:40:15 +02:00
maskformer update ruff version (#30932) 2024-05-22 06:40:15 +02:00
mbart update ruff version (#30932) 2024-05-22 06:40:15 +02:00
mbart50 Adds pretrained IDs directly in the tests (#29534) 2024-03-13 14:53:27 +01:00
mega Remove static pretrained maps from the library's internals (#29112) 2024-03-25 10:33:38 +01:00
megatron_bert update ruff version (#30932) 2024-05-22 06:40:15 +02:00
megatron_gpt2 Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
mgp_str update ruff version (#30932) 2024-05-22 06:40:15 +02:00
mistral [Port] TensorFlow implementation of Mistral (#29708) 2024-05-23 17:48:49 +01:00
mixtral update ruff version (#30932) 2024-05-22 06:40:15 +02:00
mluke Adds pretrained IDs directly in the tests (#29534) 2024-03-13 14:53:27 +01:00
mobilebert update ruff version (#30932) 2024-05-22 06:40:15 +02:00
mobilenet_v1 update ruff version (#30932) 2024-05-22 06:40:15 +02:00
mobilenet_v2 update ruff version (#30932) 2024-05-22 06:40:15 +02:00
mobilevit update ruff version (#30932) 2024-05-22 06:40:15 +02:00
mobilevitv2 update ruff version (#30932) 2024-05-22 06:40:15 +02:00
mpnet update ruff version (#30932) 2024-05-22 06:40:15 +02:00
mpt Update 4 `MptIntegrationTests` expected outputs (#30989) 2024-05-23 18:27:54 +02:00
mra update ruff version (#30932) 2024-05-22 06:40:15 +02:00
mt5 Fix accelerate failing tests (#30836) 2024-05-23 17:18:58 +02:00
musicgen update ruff version (#30932) 2024-05-22 06:40:15 +02:00
musicgen_melody update ruff version (#30932) 2024-05-22 06:40:15 +02:00
mvp update ruff version (#30932) 2024-05-22 06:40:15 +02:00
nat update ruff version (#30932) 2024-05-22 06:40:15 +02:00
nezha Remove static pretrained maps from the library's internals (#29112) 2024-03-25 10:33:38 +01:00
nllb Remove deprecated properties in tokenization_nllb.py and tokenization_nllb_fast.py (#29834) 2024-05-23 18:53:26 +02:00
nllb_moe update ruff version (#30932) 2024-05-22 06:40:15 +02:00
nougat Adds pretrained IDs directly in the tests (#29534) 2024-03-13 14:53:27 +01:00
nystromformer update ruff version (#30932) 2024-05-22 06:40:15 +02:00
olmo update ruff version (#30932) 2024-05-22 06:40:15 +02:00
oneformer update ruff version (#30932) 2024-05-22 06:40:15 +02:00
openai Remove static pretrained maps from the library's internals (#29112) 2024-03-25 10:33:38 +01:00
opt update ruff version (#30932) 2024-05-22 06:40:15 +02:00
owlv2 update ruff version (#30932) 2024-05-22 06:40:15 +02:00
owlvit update ruff version (#30932) 2024-05-22 06:40:15 +02:00
paligemma Paligemma causal attention mask (#30967) 2024-05-22 19:37:15 +02:00
patchtsmixer update ruff version (#30932) 2024-05-22 06:40:15 +02:00
patchtst update ruff version (#30932) 2024-05-22 06:40:15 +02:00
pegasus update ruff version (#30932) 2024-05-22 06:40:15 +02:00
pegasus_x update ruff version (#30932) 2024-05-22 06:40:15 +02:00
perceiver Perceiver interpolate position embedding (#30979) 2024-05-24 11:13:58 +01:00
persimmon update ruff version (#30932) 2024-05-22 06:40:15 +02:00
phi update ruff version (#30932) 2024-05-22 06:40:15 +02:00
phi3 update ruff version (#30932) 2024-05-22 06:40:15 +02:00
phobert Adds pretrained IDs directly in the tests (#29534) 2024-03-13 14:53:27 +01:00
pix2struct update ruff version (#30932) 2024-05-22 06:40:15 +02:00
plbart update ruff version (#30932) 2024-05-22 06:40:15 +02:00
poolformer update ruff version (#30932) 2024-05-22 06:40:15 +02:00
pop2piano update ruff version (#30932) 2024-05-22 06:40:15 +02:00
prophetnet update ruff version (#30932) 2024-05-22 06:40:15 +02:00
pvt update ruff version (#30932) 2024-05-22 06:40:15 +02:00
pvt_v2 Remove static pretrained maps from the library's internals (#29112) 2024-03-25 10:33:38 +01:00
qdqbert update ruff version (#30932) 2024-05-22 06:40:15 +02:00
qwen2 update ruff version (#30932) 2024-05-22 06:40:15 +02:00
qwen2_moe update ruff version (#30932) 2024-05-22 06:40:15 +02:00
rag Add `dataset_revision` argument to `RagConfig` (#29610) 2024-03-14 16:48:11 +01:00
realm update ruff version (#30932) 2024-05-22 06:40:15 +02:00
recurrent_gemma update ruff version (#30932) 2024-05-22 06:40:15 +02:00
reformer Do not remove half seq length in generation tests (#30016) 2024-04-19 17:32:52 +01:00
regnet update ruff version (#30932) 2024-05-22 06:40:15 +02:00
rembert update ruff version (#30932) 2024-05-22 06:40:15 +02:00
resnet update ruff version (#30932) 2024-05-22 06:40:15 +02:00
roberta Remove static pretrained maps from the library's internals (#29112) 2024-03-25 10:33:38 +01:00
roberta_prelayernorm Remove static pretrained maps from the library's internals (#29112) 2024-03-25 10:33:38 +01:00
roc_bert update ruff version (#30932) 2024-05-22 06:40:15 +02:00
roformer update ruff version (#30932) 2024-05-22 06:40:15 +02:00
rwkv Remove static pretrained maps from the library's internals (#29112) 2024-03-25 10:33:38 +01:00
sam update ruff version (#30932) 2024-05-22 06:40:15 +02:00
seamless_m4t Remove deprecated properties in tokenization_nllb.py and tokenization_nllb_fast.py (#29834) 2024-05-23 18:53:26 +02:00
seamless_m4t_v2 update ruff version (#30932) 2024-05-22 06:40:15 +02:00
segformer update ruff version (#30932) 2024-05-22 06:40:15 +02:00
seggpt update ruff version (#30932) 2024-05-22 06:40:15 +02:00
sew update ruff version (#30932) 2024-05-22 06:40:15 +02:00
sew_d update ruff version (#30932) 2024-05-22 06:40:15 +02:00
siglip Fix accelerate failing tests (#30836) 2024-05-23 17:18:58 +02:00
speech_encoder_decoder [tests] add `torch.use_deterministic_algorithms` for XPU (#30774) 2024-05-23 16:53:07 +01:00
speech_to_text update ruff version (#30932) 2024-05-22 06:40:15 +02:00
speech_to_text_2 update ruff version (#30932) 2024-05-22 06:40:15 +02:00
speecht5 [tests] add `torch.use_deterministic_algorithms` for XPU (#30774) 2024-05-23 16:53:07 +01:00
splinter update ruff version (#30932) 2024-05-22 06:40:15 +02:00
squeezebert Remove static pretrained maps from the library's internals (#29112) 2024-03-25 10:33:38 +01:00
stablelm update ruff version (#30932) 2024-05-22 06:40:15 +02:00
starcoder2 update ruff version (#30932) 2024-05-22 06:40:15 +02:00
superpoint Removal of deprecated maps (#30576) 2024-05-09 14:15:56 +02:00
swiftformer update ruff version (#30932) 2024-05-22 06:40:15 +02:00
swin update ruff version (#30932) 2024-05-22 06:40:15 +02:00
swin2sr update ruff version (#30932) 2024-05-22 06:40:15 +02:00
swinv2 update ruff version (#30932) 2024-05-22 06:40:15 +02:00
switch_transformers Fix accelerate failing tests (#30836) 2024-05-23 17:18:58 +02:00
t5 Fix accelerate failing tests (#30836) 2024-05-23 17:18:58 +02:00
table_transformer update ruff version (#30932) 2024-05-22 06:40:15 +02:00
tapas update ruff version (#30932) 2024-05-22 06:40:15 +02:00
time_series_transformer update ruff version (#30932) 2024-05-22 06:40:15 +02:00
timesformer update ruff version (#30932) 2024-05-22 06:40:15 +02:00
timm_backbone 🚨 out_indices always a list (#30941) 2024-05-22 15:23:04 +01:00
trocr update ruff version (#30932) 2024-05-22 06:40:15 +02:00
tvlt update ruff version (#30932) 2024-05-22 06:40:15 +02:00
tvp update ruff version (#30932) 2024-05-22 06:40:15 +02:00
udop Add split special tokens (#30772) 2024-05-24 08:38:58 -07:00
umt5 Fix accelerate failing tests (#30836) 2024-05-23 17:18:58 +02:00
unispeech update ruff version (#30932) 2024-05-22 06:40:15 +02:00
unispeech_sat update ruff version (#30932) 2024-05-22 06:40:15 +02:00
univnet Add tests for batching support (#29297) 2024-03-12 17:46:19 +00:00
upernet update ruff version (#30932) 2024-05-22 06:40:15 +02:00
video_llava update ruff version (#30932) 2024-05-22 06:40:15 +02:00
videomae update ruff version (#30932) 2024-05-22 06:40:15 +02:00
vilt update ruff version (#30932) 2024-05-22 06:40:15 +02:00
vipllava update ruff version (#30932) 2024-05-22 06:40:15 +02:00
vision_encoder_decoder update ruff version (#30932) 2024-05-22 06:40:15 +02:00
vision_text_dual_encoder update ruff version (#30932) 2024-05-22 06:40:15 +02:00
visual_bert update ruff version (#30932) 2024-05-22 06:40:15 +02:00
vit update ruff version (#30932) 2024-05-22 06:40:15 +02:00
vit_hybrid update ruff version (#30932) 2024-05-22 06:40:15 +02:00
vit_mae added interpolation for vitmae model in pytorch as well as tf. (#30732) 2024-05-24 16:20:09 +01:00
vit_msn update ruff version (#30932) 2024-05-22 06:40:15 +02:00
vitdet update ruff version (#30932) 2024-05-22 06:40:15 +02:00
vitmatte update ruff version (#30932) 2024-05-22 06:40:15 +02:00
vits update ruff version (#30932) 2024-05-22 06:40:15 +02:00
vivit update ruff version (#30932) 2024-05-22 06:40:15 +02:00
wav2vec2 update ruff version (#30932) 2024-05-22 06:40:15 +02:00
wav2vec2_bert update ruff version (#30932) 2024-05-22 06:40:15 +02:00
wav2vec2_conformer update ruff version (#30932) 2024-05-22 06:40:15 +02:00
wav2vec2_phoneme update ruff version (#30932) 2024-05-22 06:40:15 +02:00
wav2vec2_with_lm Add AutoFeatureExtractor support to Wav2Vec2ProcessorWithLM (#28706) 2024-05-20 13:40:42 +02:00
wavlm update ruff version (#30932) 2024-05-22 06:40:15 +02:00
whisper Fix pad_to_max_length Whisper (#30787) 2024-05-27 16:09:05 +02:00
x_clip update ruff version (#30932) 2024-05-22 06:40:15 +02:00
xglm Remove static pretrained maps from the library's internals (#29112) 2024-03-25 10:33:38 +01:00
xlm Fix 29807 sinusoidal positional encodings in Flaubert, Informer and XLM (#29904) 2024-04-02 10:27:26 +02:00
xlm_prophetnet Adds pretrained IDs directly in the tests (#29534) 2024-03-13 14:53:27 +01:00
xlm_roberta Adds pretrained IDs directly in the tests (#29534) 2024-03-13 14:53:27 +01:00
xlm_roberta_xl Revert low cpu mem tie weights (#29135) 2024-02-20 12:06:46 +00:00
xlnet update ruff version (#30932) 2024-05-22 06:40:15 +02:00
xmod Update all references to canonical models (#29001) 2024-02-16 08:16:58 +01:00
yolos update ruff version (#30932) 2024-05-22 06:40:15 +02:00
yoso update ruff version (#30932) 2024-05-22 06:40:15 +02:00
__init__.py Move test model folders (#17034) 2022-05-03 14:42:02 +02:00