saymrwulf/transformers
1
0
Fork 0
mirror of https://github.com/saymrwulf/transformers.git synced 2026-05-14 20:58:08 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions
transformers/tests/models
History
NielsRogge 9a6c6ef97f
[Backbones] Improve out features (#20675) 
* Improve ResNet backbone

* Improve Bit backbone

* Improve docstrings

* Fix default stage

* Apply suggestions from code review

Co-authored-by: Niels Rogge <nielsrogge@Nielss-MacBook-Pro.local>
2022-12-09 09:14:52 +01:00
..
albert Improve model tester (#19984) 2022-11-02 17:38:44 +01:00
audio_spectrogram_transformer Add Audio Spectogram Transformer (#19981) 2022-11-21 18:58:54 +01:00
auto Fix test for file not found (#20604) 2022-12-05 18:33:56 -05:00
bart Add TFBartForSequenceClassification (#20570) 2022-12-07 18:05:39 +01:00
barthez
bartpho
beit Add Image Processors (#19796) 2022-11-02 11:57:36 +00:00
bert update relative positional embedding (#20203) 2022-11-15 10:46:34 +01:00
bert_generation Generate: move generation_*.py src files into generation/*.py (#20096) 2022-11-09 15:34:08 +00:00
bert_japanese Add sentencepiece to BertJapaneseTokenizer (#19769) 2022-10-21 10:04:49 -04:00
bertweet
big_bird wrap forward passes with torch.no_grad() (#19273) 2022-10-04 16:13:22 +02:00
bigbird_pegasus Generate: move generation_*.py src files into generation/*.py (#20096) 2022-11-09 15:34:08 +00:00
biogpt Add BioGPT (#20420) 2022-12-05 10:12:03 -05:00
bit [Backbones] Improve out features (#20675) 2022-12-09 09:14:52 +01:00
blenderbot Generate: move generation_*.py src files into generation/*.py (#20096) 2022-11-09 15:34:08 +00:00
blenderbot_small Generate: move generation_*.py src files into generation/*.py (#20096) 2022-11-09 15:34:08 +00:00
bloom Generate: move generation_*.py src files into generation/*.py (#20096) 2022-11-09 15:34:08 +00:00
bort
byt5
camembert
canine Improve model tester (#19984) 2022-11-02 17:38:44 +01:00
chinese_clip Add Chinese-CLIP implementation (#20368) 2022-11-30 19:22:23 +01:00
clip [CLIP] allow loading projection layer in vision and text model (#18962) 2022-11-15 17:50:07 +01:00
clipseg Fix device issues in CLIPSegModelIntegrationTest (#20467) 2022-11-28 10:41:28 +01:00
codegen Generate: move generation_*.py src files into generation/*.py (#20096) 2022-11-09 15:34:08 +00:00
conditional_detr Add segmentation + object detection image processors (#20160) 2022-11-30 10:24:03 +00:00
convbert wrap forward passes with torch.no_grad() (#19274) 2022-10-04 16:12:03 +02:00
convnext Add Image Processors (#19796) 2022-11-02 11:57:36 +00:00
cpm
ctrl Generate: move generation_*.py src files into generation/*.py (#20096) 2022-11-09 15:34:08 +00:00
cvt [CvT] Tensorflow implementation (#18597) 2022-10-11 18:16:52 +01:00
data2vec update relative positional embedding (#20203) 2022-11-15 10:46:34 +01:00
deberta
deberta_v2 🚨 🚨 🚨 Fix Issue 15003: SentencePiece Tokenizers Not Adding Special Tokens in convert_tokens_to_string (#15775) 2022-11-02 15:45:38 -04:00
decision_transformer Generate: move generation_*.py src files into generation/*.py (#20096) 2022-11-09 15:34:08 +00:00
deformable_detr Add segmentation + object detection image processors (#20160) 2022-11-30 10:24:03 +00:00
deit [Vision] .to function for ImageProcessors (#20536) 2022-12-05 19:10:54 +01:00
detr Add segmentation + object detection image processors (#20160) 2022-11-30 10:24:03 +00:00
dinat Add Neighborhood Attention Transformer (NAT) and Dilated NAT (DiNAT) models (#20219) 2022-11-18 13:08:26 -05:00
distilbert
dit
donut Add Donut image processor (#20425) 2022-11-29 10:38:01 +00:00
dpr
dpt Add dpt-hybrid support (#20645) 2022-12-07 17:01:55 +01:00
electra Improve model tester (#19984) 2022-11-02 17:38:44 +01:00
encoder_decoder Generate: move generation_*.py src files into generation/*.py (#20096) 2022-11-09 15:34:08 +00:00
ernie update relative positional embedding (#20203) 2022-11-15 10:46:34 +01:00
esm Fix expected values for TF-ESM tests (#20680) 2022-12-08 15:26:09 +00:00
flaubert Improve model tester (#19984) 2022-11-02 17:38:44 +01:00
flava Fix type - update any PIL.Image.Resampling (#20172) 2022-11-11 16:55:59 +00:00
fnet wrap forward passes with torch.no_grad() (#19413) 2022-10-10 15:03:46 -04:00
fsmt Generate: move generation_*.py src files into generation/*.py (#20096) 2022-11-09 15:34:08 +00:00
funnel
glpn
gpt2 Fix TF nightly tests (#20507) 2022-11-30 14:47:54 +00:00
gpt_neo Generate: move generation_*.py src files into generation/*.py (#20096) 2022-11-09 15:34:08 +00:00
gpt_neox
gpt_neox_japanese Add support for Japanese GPT-NeoX-based model by ABEJA, Inc. (#18814) 2022-09-14 10:17:40 -04:00
gptj Fix flax GPT-J-6B linking model in tests (#20556) 2022-12-05 14:00:05 +01:00
groupvit Allow passing arguments to model testers for CLIP-like models (#20044) 2022-11-04 18:01:41 +01:00
herbert
hubert Fix train_step, test_step and tests for CLIP (#18684) 2022-09-09 20:01:02 +01:00
ibert Improve model tester (#19984) 2022-11-02 17:38:44 +01:00
imagegpt Generate: move generation_*.py src files into generation/*.py (#20096) 2022-11-09 15:34:08 +00:00
jukebox Ci-jukebox (#20613) 2022-12-06 16:14:03 +01:00
layoutlm TF: TF 2.10 unpin + related onnx test skips (#18995) 2022-09-12 19:30:27 +01:00
layoutlmv2 [processor] Add 'model input names' property (#20117) 2022-11-10 19:29:20 +00:00
layoutlmv3 [processor] Add 'model input names' property (#20117) 2022-11-10 19:29:20 +00:00
layoutxlm Fix tapas scatter (#20149) 2022-11-14 01:04:26 -05:00
led Added missing test_tokenization_led (#20568) 2022-12-08 20:55:22 +01:00
levit Add Image Processors (#19796) 2022-11-02 11:57:36 +00:00
lilt Generate: move generation_*.py src files into generation/*.py (#20096) 2022-11-09 15:34:08 +00:00
longformer Improve model tester (#19984) 2022-11-02 17:38:44 +01:00
longt5 Generate: move generation_*.py src files into generation/*.py (#20096) 2022-11-09 15:34:08 +00:00
luke
lxmert
m2m_100 Generate: move generation_*.py src files into generation/*.py (#20096) 2022-11-09 15:34:08 +00:00
marian Generate: move generation_*.py src files into generation/*.py (#20096) 2022-11-09 15:34:08 +00:00
markuplm [processor] Add 'model input names' property (#20117) 2022-11-10 19:29:20 +00:00
maskformer [MaskFormer] Add support for ResNet backbone (#20483) 2022-12-07 09:42:38 +01:00
mbart Enable PyTorch 1.13 (#20168) 2022-11-15 11:33:09 -05:00
mbart50
mctct [processor] Add 'model input names' property (#20117) 2022-11-10 19:29:20 +00:00
megatron_bert
megatron_gpt2
mluke
mobilebert Fix train_step, test_step and tests for CLIP (#18684) 2022-09-09 20:01:02 +01:00
mobilenet_v1 add MobileNetV1 model (#17799) 2022-11-21 10:21:28 -05:00
mobilenet_v2 add MobileNetV2 model (#17845) 2022-11-14 01:00:10 -05:00
mobilevit Add Image Processors (#19796) 2022-11-02 11:57:36 +00:00
mpnet
mt5
mvp Generate: move generation_*.py src files into generation/*.py (#20096) 2022-11-09 15:34:08 +00:00
nat Add Neighborhood Attention Transformer (NAT) and Dilated NAT (DiNAT) models (#20219) 2022-11-18 13:08:26 -05:00
nezha Generate: move generation_*.py src files into generation/*.py (#20096) 2022-11-09 15:34:08 +00:00
nllb
nystromformer
openai Generate: move generation_*.py src files into generation/*.py (#20096) 2022-11-09 15:34:08 +00:00
opt Generate: move generation_*.py src files into generation/*.py (#20096) 2022-11-09 15:34:08 +00:00
owlvit Add segmentation + object detection image processors (#20160) 2022-11-30 10:24:03 +00:00
pegasus Generate: move generation_*.py src files into generation/*.py (#20096) 2022-11-09 15:34:08 +00:00
pegasus_x Generate: move generation_*.py src files into generation/*.py (#20096) 2022-11-09 15:34:08 +00:00
perceiver Update defaults and logic to match old FE (#20065) 2022-11-04 19:14:56 +00:00
phobert
plbart Generate: fix plbart generation tests (#20391) 2022-11-22 17:56:04 +00:00
poolformer Add Image Processors (#19796) 2022-11-02 11:57:36 +00:00
prophetnet Generate: move generation_*.py src files into generation/*.py (#20096) 2022-11-09 15:34:08 +00:00
qdqbert
rag
realm
reformer Generate: move generation_*.py src files into generation/*.py (#20096) 2022-11-09 15:34:08 +00:00
regnet Run tests if skip condition not met (#18764) 2022-08-30 14:03:28 +02:00
rembert
resnet [Backbones] Improve out features (#20675) 2022-12-09 09:14:52 +01:00
retribert
roberta update relative positional embedding (#20203) 2022-11-15 10:46:34 +01:00
roc_bert fix both failing RoCBert tests (#20469) 2022-11-28 17:08:57 +01:00
roformer wrap forward passes with torch.no_grad() (#19438) 2022-10-10 14:54:54 -04:00
segformer Add Image Processors (#19796) 2022-11-02 11:57:36 +00:00
sew
sew_d
speech_encoder_decoder Generate: move generation_*.py src files into generation/*.py (#20096) 2022-11-09 15:34:08 +00:00
speech_to_text [processor] Add 'model input names' property (#20117) 2022-11-10 19:29:20 +00:00
speech_to_text_2 Generate: move generation_*.py src files into generation/*.py (#20096) 2022-11-09 15:34:08 +00:00
splinter
squeezebert
swin remvoe _create_and_check_torch_fx_tracing in specific test files (#18667) 2022-09-07 16:22:09 +02:00
swinv2
switch_transformers [Switch Transformers] Fix failing slow test (#20346) 2022-11-21 15:36:49 +01:00
t5 [FLAX] Add dtype to embedding for bert/bart/opt/t5 (#20340) 2022-11-28 10:21:42 -05:00
table_transformer Generate: move generation_*.py src files into generation/*.py (#20096) 2022-11-09 15:34:08 +00:00
tapas Fix tapas scatter (#20149) 2022-11-14 01:04:26 -05:00
tapex
time_series_transformer Add a decorator for flaky tests (#19498) 2022-10-12 14:00:17 -04:00
timesformer [New Model] Add TimeSformer model (#18908) 2022-12-02 09:13:25 +01:00
trajectory_transformer Generate: move generation_*.py src files into generation/*.py (#20096) 2022-11-09 15:34:08 +00:00
transfo_xl Generate: move generation_*.py src files into generation/*.py (#20096) 2022-11-09 15:34:08 +00:00
trocr Generate: move generation_*.py src files into generation/*.py (#20096) 2022-11-09 15:34:08 +00:00
unispeech
unispeech_sat
van
videomae Add Image Processors (#19796) 2022-11-02 11:57:36 +00:00
vilt Add Image Processors (#19796) 2022-11-02 11:57:36 +00:00
vision_encoder_decoder Fix donut image processor (#20625) 2022-12-08 19:10:40 +00:00
vision_text_dual_encoder [processor] Add 'model input names' property (#20117) 2022-11-10 19:29:20 +00:00
visual_bert wrap forward passes with torch.no_grad() (#19439) 2022-10-10 14:54:36 -04:00
vit Add accelerate support for ViT family (#20174) 2022-11-15 11:06:01 +01:00
vit_hybrid [ViTHybrid] Fix accelerate slow tests (#20679) 2022-12-08 17:39:32 +01:00
vit_mae TF: tests for (de)serializable models with resized tokens (#19013) 2022-09-16 16:38:08 +01:00
vit_msn Some fixes regarding auto mappings and test class names (#19923) 2022-10-27 14:38:59 +02:00
wav2vec2 [processor] Add 'model input names' property (#20117) 2022-11-10 19:29:20 +00:00
wav2vec2_conformer
wav2vec2_phoneme
wav2vec2_with_lm [processor] Add 'model input names' property (#20117) 2022-11-10 19:29:20 +00:00
wavlm
whisper [Whisper] Fix forced decoder ids (#20652) 2022-12-07 16:44:13 +00:00
x_clip Allow passing arguments to model testers for CLIP-like models (#20044) 2022-11-04 18:01:41 +01:00
xglm Generate: move generation_*.py src files into generation/*.py (#20096) 2022-11-09 15:34:08 +00:00
xlm Generate: move generation_*.py src files into generation/*.py (#20096) 2022-11-09 15:34:08 +00:00
xlm_prophetnet
xlm_roberta
xlm_roberta_xl update relative positional embedding (#20203) 2022-11-15 10:46:34 +01:00
xlnet Generate: move generation_*.py src files into generation/*.py (#20096) 2022-11-09 15:34:08 +00:00
yolos Add segmentation + object detection image processors (#20160) 2022-11-30 10:24:03 +00:00
yoso
__init__.py