| .. |
|
albert
|
Improve model tester (#19984)
|
2022-11-02 17:38:44 +01:00 |
|
auto
|
Fix tapas scatter (#20149)
|
2022-11-14 01:04:26 -05:00 |
|
bart
|
Generate: move generation_*.py src files into generation/*.py (#20096)
|
2022-11-09 15:34:08 +00:00 |
|
barthez
|
|
|
|
bartpho
|
|
|
|
beit
|
Add Image Processors (#19796)
|
2022-11-02 11:57:36 +00:00 |
|
bert
|
Generate: move generation_*.py src files into generation/*.py (#20096)
|
2022-11-09 15:34:08 +00: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 |
|
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
|
Replace as_target context managers by direct calls (#18325)
|
2022-07-29 08:09:09 -04:00 |
|
camembert
|
|
|
|
canine
|
Improve model tester (#19984)
|
2022-11-02 17:38:44 +01:00 |
|
clip
|
[processor] Add 'model input names' property (#20117)
|
2022-11-10 19:29:20 +00:00 |
|
clipseg
|
Update CLIPSegModelTester (#20134)
|
2022-11-09 15:21:52 +01:00 |
|
codegen
|
Generate: move generation_*.py src files into generation/*.py (#20096)
|
2022-11-09 15:34:08 +00:00 |
|
conditional_detr
|
Generate: move generation_*.py src files into generation/*.py (#20096)
|
2022-11-09 15:34:08 +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
|
Generate: move generation_*.py src files into generation/*.py (#20096)
|
2022-11-09 15:34:08 +00: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
|
Generate: move generation_*.py src files into generation/*.py (#20096)
|
2022-11-09 15:34:08 +00:00 |
|
deit
|
Add Image Processors (#19796)
|
2022-11-02 11:57:36 +00:00 |
|
detr
|
Generate: move generation_*.py src files into generation/*.py (#20096)
|
2022-11-09 15:34:08 +00:00 |
|
distilbert
|
|
|
|
dit
|
|
|
|
donut
|
remvoe _create_and_check_torch_fx_tracing in specific test files (#18667)
|
2022-09-07 16:22:09 +02:00 |
|
dpr
|
Replace as_target context managers by direct calls (#18325)
|
2022-07-29 08:09:09 -04:00 |
|
dpt
|
Add Image Processors (#19796)
|
2022-11-02 11:57:36 +00: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
|
Generate: move generation_*.py src files into generation/*.py (#20096)
|
2022-11-09 15:34:08 +00:00 |
|
esm
|
Fix ESM LM head test (#20045)
|
2022-11-04 12:45:34 +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
|
Update serving code to enable saved_model=True (#18153)
|
2022-07-22 18:05:38 +01:00 |
|
glpn
|
|
|
|
gpt2
|
Generate: move generation_*.py src files into generation/*.py (#20096)
|
2022-11-09 15:34:08 +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
|
Generate: move generation_*.py src files into generation/*.py (#20096)
|
2022-11-09 15:34:08 +00: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
|
Add Jukebox model (replaces #16875) (#17826)
|
2022-11-10 21:05:27 +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
|
Generate: move generation_*.py src files into generation/*.py (#20096)
|
2022-11-09 15:34:08 +00: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
|
Adding fine-tuning models to LUKE (#18353)
|
2022-08-01 11:09:47 -04:00 |
|
lxmert
|
Update serving code to enable saved_model=True (#18153)
|
2022-07-22 18:05:38 +01:00 |
|
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
|
Fix image segmentation pipeline errors, resolve backward compatibility issues (#19768)
|
2022-10-21 18:09:58 +03:00 |
|
mbart
|
Generate: move generation_*.py src files into generation/*.py (#20096)
|
2022-11-09 15:34:08 +00:00 |
|
mbart50
|
Replace as_target context managers by direct calls (#18325)
|
2022-07-29 08:09:09 -04:00 |
|
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_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
|
Fix expected loss values in some (m)T5 tests (#18177)
|
2022-07-18 15:26:21 +02:00 |
|
mvp
|
Generate: move generation_*.py src files into generation/*.py (#20096)
|
2022-11-09 15:34:08 +00:00 |
|
nezha
|
Generate: move generation_*.py src files into generation/*.py (#20096)
|
2022-11-09 15:34:08 +00:00 |
|
nllb
|
Replace as_target context managers by direct calls (#18325)
|
2022-07-29 08:09:09 -04:00 |
|
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
|
[OWL-ViT] Make model consistent with CLIP (#20144)
|
2022-11-11 11:36:17 +01: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: move generation_*.py src files into generation/*.py (#20096)
|
2022-11-09 15:34:08 +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
|
FX support for ConvNext, Wav2Vec2 and ResNet (#19053)
|
2022-09-16 10:57:41 +02:00 |
|
retribert
|
|
|
|
roberta
|
Generate: move generation_*.py src files into generation/*.py (#20096)
|
2022-11-09 15:34:08 +00:00 |
|
roc_bert
|
[ROC_BERT] Make CI happy (#20175)
|
2022-11-14 18:04:25 +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
|
Add swin transformer v2 (#17469)
|
2022-07-27 11:14:47 -04:00 |
|
t5
|
Generate: move generation_*.py src files into generation/*.py (#20096)
|
2022-11-09 15:34:08 +00: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
|
Replace as_target context managers by direct calls (#18325)
|
2022-07-29 08:09:09 -04:00 |
|
time_series_transformer
|
Add a decorator for flaky tests (#19498)
|
2022-10-12 14:00:17 -04: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
|
Generate: move generation_*.py src files into generation/*.py (#20096)
|
2022-11-09 15:34:08 +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 Image Processors (#19796)
|
2022-11-02 11:57:36 +00: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
|
[processor] Add 'model input names' property (#20117)
|
2022-11-10 19:29:20 +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
|
Generate: move generation_*.py src files into generation/*.py (#20096)
|
2022-11-09 15:34:08 +00:00 |
|
xlnet
|
Generate: move generation_*.py src files into generation/*.py (#20096)
|
2022-11-09 15:34:08 +00:00 |
|
yolos
|
[fix] Add DeformableDetrFeatureExtractor (#19140)
|
2022-09-22 09:45:24 +02:00 |
|
yoso
|
|
|
|
__init__.py
|
|
|