transformers/tests/models
Omar Salman 681183974a
Enable dynamic resolution input for Beit (#31053)
* Initial attempt

* Updates: PR suggestions

* Interpolate the relative position bias when interpolate_pos_encoding is True

* Add slow tag for the added tests

* Add in DATA2VEC_VISION_INPUTS_DOCSTRING
2024-06-06 14:47:41 +01:00
..
albert
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
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
bartpho
beit Enable dynamic resolution input for Beit (#31053) 2024-06-06 14:47:41 +01:00
bert Ignore non-causal mask in more cases with SDPA (#30138) 2024-06-03 19:08:41 +08:00
bert_generation
bert_japanese
bertweet
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
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
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
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 fix the get_size_with_aspect_ratio in max_size situation (#30902) 2024-06-03 16:12:08 +01: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
cpmant update ruff version (#30932) 2024-05-22 06:40:15 +02:00
ctrl
cvt update ruff version (#30932) 2024-05-22 06:40:15 +02:00
data2vec Enable dynamic resolution input for Beit (#31053) 2024-06-06 14:47:41 +01:00
dbrx Fix accelerate failing tests (#30836) 2024-05-23 17:18:58 +02:00
deberta
deberta_v2
decision_transformer update ruff version (#30932) 2024-05-22 06:40:15 +02:00
deformable_detr fix the get_size_with_aspect_ratio in max_size situation (#30902) 2024-06-03 16:12:08 +01:00
deit Add dynamic resolution input/interpolate position embedding to deit (#31131) 2024-06-04 10:29:01 +01:00
depth_anything update ruff version (#30932) 2024-05-22 06:40:15 +02:00
detr fix the get_size_with_aspect_ratio in max_size situation (#30902) 2024-06-03 16:12:08 +01: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
dit
donut update ruff version (#30932) 2024-05-22 06:40:15 +02:00
dpr
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
esm Fix sentence fragment within test comments (#31218) 2024-06-04 10:09:24 +01: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 Add missing Flaubert tokenizer tests (#30492) 2024-06-05 13:52:16 +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
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
gptj
grounding_dino fix the get_size_with_aspect_ratio in max_size situation (#30902) 2024-06-03 16:12:08 +01:00
groupvit update ruff version (#30932) 2024-05-22 06:40:15 +02:00
herbert
hubert update ruff version (#30932) 2024-05-22 06:40:15 +02:00
ibert Fix sentence fragment within test comments (#31218) 2024-06-04 10:09:24 +01: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 Skip failing JetMOE generation tests (#31266) 2024-06-05 19:06:46 +01: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
llama Fix test_compile_static_cache (#30991) 2024-06-03 15:16:28 +02: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 Make mamba use cache (#31116) 2024-06-06 13:37:29 +05: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
megatron_bert update ruff version (#30932) 2024-05-22 06:40:15 +02:00
megatron_gpt2
mgp_str update ruff version (#30932) 2024-05-22 06:40:15 +02:00
mistral Fix MistralIntegrationTest (#31231) 2024-06-04 18:04:08 +02:00
mixtral update ruff version (#30932) 2024-05-22 06:40:15 +02:00
mluke
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
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
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
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 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
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
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
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 Fix sentence fragment within test comments (#31218) 2024-06-04 10:09:24 +01:00
roberta_prelayernorm Fix sentence fragment within test comments (#31218) 2024-06-04 10:09:24 +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 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 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
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
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
upernet update ruff version (#30932) 2024-05-22 06:40:15 +02:00
video_llava Video-LLaVa: handle any number of frames (#31221) 2024-06-04 14:20:03 +05: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
xlm
xlm_roberta
xlm_roberta_xl fix accelerate tests for roberta xl (#31288) 2024-06-06 14:44:35 +01:00
xlnet update ruff version (#30932) 2024-05-22 06:40:15 +02:00
xmod Fix sentence fragment within test comments (#31218) 2024-06-04 10:09:24 +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