transformers/tests/models
Matt c9e4f75f9b Remove ConversationalPipeline and Conversation object, as they have been deprecated for some time and are due for removal 2024-05-31 14:29:45 +01: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 Remove ConversationalPipeline and Conversation object, as they have been deprecated for some time and are due for removal 2024-05-31 14:29:45 +01: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 Remove ConversationalPipeline and Conversation object, as they have been deprecated for some time and are due for removal 2024-05-31 14:29:45 +01: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 Remove ConversationalPipeline and Conversation object, as they have been deprecated for some time and are due for removal 2024-05-31 14:29:45 +01:00
blenderbot_small Remove ConversationalPipeline and Conversation object, as they have been deprecated for some time and are due for removal 2024-05-31 14:29:45 +01: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 Fix failing tokenizer tests (#31083) 2024-05-28 13:34:23 +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 Remove `ninja` from docker image build (#31080) 2024-05-28 11:36:26 +02:00
deit skip `test_multi_gpu_data_parallel_forward` for `vit` and `deit` (#31086) 2024-05-28 17:44:52 +02:00
depth_anything 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
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
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 Remove ConversationalPipeline and Conversation object, as they have been deprecated for some time and are due for removal 2024-05-31 14:29:45 +01: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
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 TST: Fix instruct-blip tests (#31088) 2024-05-28 18:29:11 +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
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 Remove ConversationalPipeline and Conversation object, as they have been deprecated for some time and are due for removal 2024-05-31 14:29:45 +01: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 Remove ConversationalPipeline and Conversation object, as they have been deprecated for some time and are due for removal 2024-05-31 14:29:45 +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 Remove ConversationalPipeline and Conversation object, as they have been deprecated for some time and are due for removal 2024-05-31 14:29:45 +01:00
mamba Mamba `slow_forward` gradient fix (#29563) 2024-03-27 04:52:12 +01:00
marian Remove ConversationalPipeline and Conversation object, as they have been deprecated for some time and are due for removal 2024-05-31 14:29:45 +01: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 Remove ConversationalPipeline and Conversation object, as they have been deprecated for some time and are due for removal 2024-05-31 14:29:45 +01:00
mbart50 Adds pretrained IDs directly in the tests (#29534) 2024-03-13 14:53:27 +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 Remove ConversationalPipeline and Conversation object, as they have been deprecated for some time and are due for removal 2024-05-31 14:29:45 +01: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 Remove ConversationalPipeline and Conversation object, as they have been deprecated for some time and are due for removal 2024-05-31 14:29:45 +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 Remove ConversationalPipeline and Conversation object, as they have been deprecated for some time and are due for removal 2024-05-31 14:29:45 +01: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 Fix OWLv2 post_process_object_detection for multiple images (#31082) 2024-05-28 12:06:06 +01: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 Remove ConversationalPipeline and Conversation object, as they have been deprecated for some time and are due for removal 2024-05-31 14:29:45 +01:00
pegasus_x Remove ConversationalPipeline and Conversation object, as they have been deprecated for some time and are due for removal 2024-05-31 14:29:45 +01: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 Remove ConversationalPipeline and Conversation object, as they have been deprecated for some time and are due for removal 2024-05-31 14:29:45 +01: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 Remove ConversationalPipeline and Conversation object, as they have been deprecated for some time and are due for removal 2024-05-31 14:29:45 +01: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
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
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 `ninja` from docker image build (#31080) 2024-05-28 11:36:26 +02:00
sam update ruff version (#30932) 2024-05-22 06:40:15 +02:00
seamless_m4t Remove ConversationalPipeline and Conversation object, as they have been deprecated for some time and are due for removal 2024-05-31 14:29:45 +01: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 Deprecate low use models (#30781) 2024-05-28 18:07:07 +01:00
speech_to_text 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 Remove ConversationalPipeline and Conversation object, as they have been deprecated for some time and are due for removal 2024-05-31 14:29:45 +01:00
t5 Remove ConversationalPipeline and Conversation object, as they have been deprecated for some time and are due for removal 2024-05-31 14:29:45 +01: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
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 Remove ConversationalPipeline and Conversation object, as they have been deprecated for some time and are due for removal 2024-05-31 14:29:45 +01: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 skip `test_multi_gpu_data_parallel_forward` for `vit` and `deit` (#31086) 2024-05-28 17:44:52 +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_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