SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity
Files
8a1a23ae4dfb022a2e483506db973ceed41f5fac
HuggingFace_transformer/tests/models
History
Yih-Dar 8a1a23ae4d Fix GPU OOM for mistral.py::Mask4DTestHard (#31212) 
* build

* build

* build

* build

---------

Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
2024-06-03 19:25:15 +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
Ignore non-causal mask in more cases with SDPA (#30138)
2024-06-03 19:08:41 +08: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
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
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
fix the get_size_with_aspect_ratio in max_size situation (#30902)
2024-06-03 16:12:08 +01: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
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
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
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
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
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
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
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
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
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
Fix GPU OOM for mistral.py::Mask4DTestHard (#31212)
2024-06-03 19:25:15 +02: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
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
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
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
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 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
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
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
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