transformers/tests/models
Carl d6b8e9cec7
Add trajectory transformer (#17141)
* Add trajectory transformer


Fix model init


Fix end of lines for .mdx files

Add trajectory transformer model to toctree

Add forward input docs

Fix docs, remove prints, simplify prediction test

Apply suggestions from code review

Co-authored-by: Sylvain Gugger <35901082+sgugger@users.noreply.github.com>
Apply suggestions from code review

Co-authored-by: Lysandre Debut <lysandre@huggingface.co>
Co-authored-by: Sylvain Gugger <35901082+sgugger@users.noreply.github.com>
Update docs, more descriptive comments

Apply suggestions from code review

Co-authored-by: Sylvain Gugger <35901082+sgugger@users.noreply.github.com>
Update readme

Small comment update and add conversion script

Rebase and reformat

Fix copies

Fix rebase, remove duplicates

Fix rebase, remove duplicates

* Remove tapex

* Remove tapex

* Remove tapex
2022-05-17 19:07:43 -04:00
..
albert Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
auto Automatically sort auto mappings (#17250) 2022-05-16 13:24:20 -04:00
bart Black preview (#17217) 2022-05-12 16:25:55 -04:00
barthez Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
bartpho Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
beit Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
bert Black preview (#17217) 2022-05-12 16:25:55 -04:00
bert_generation Black preview (#17217) 2022-05-12 16:25:55 -04:00
bert_japanese Black preview (#17217) 2022-05-12 16:25:55 -04:00
bertweet Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
big_bird Black preview (#17217) 2022-05-12 16:25:55 -04:00
bigbird_pegasus Black preview (#17217) 2022-05-12 16:25:55 -04:00
blenderbot Black preview (#17217) 2022-05-12 16:25:55 -04:00
blenderbot_small Black preview (#17217) 2022-05-12 16:25:55 -04:00
bort Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
byt5 Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
camembert Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
canine Black preview (#17217) 2022-05-12 16:25:55 -04:00
clip Convert image to rgb for clip model (#17101) 2022-05-11 13:09:54 +01:00
convbert Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
convnext Black preview (#17217) 2022-05-12 16:25:55 -04:00
cpm Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
ctrl Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
data2vec Black preview (#17217) 2022-05-12 16:25:55 -04:00
deberta Black preview (#17217) 2022-05-12 16:25:55 -04:00
deberta_v2 Add DebertaV2ForMultipleChoice (#17135) 2022-05-10 16:21:44 -04:00
decision_transformer Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
deit Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
detr Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
distilbert Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
dit Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
dpr Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
dpt Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
electra Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
encoder_decoder Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
flaubert Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
flava Fix FlavaForPreTrainingIntegrationTest CI test (#17232) 2022-05-16 21:14:25 +02:00
fnet Black preview (#17217) 2022-05-12 16:25:55 -04:00
fsmt Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
funnel Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
glpn Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
gpt_neo Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
gpt2 Black preview (#17217) 2022-05-12 16:25:55 -04:00
gptj Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
herbert Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
hubert Black preview (#17217) 2022-05-12 16:25:55 -04:00
ibert Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
imagegpt Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
layoutlm Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
layoutlmv2 Black preview (#17217) 2022-05-12 16:25:55 -04:00
layoutxlm Black preview (#17217) 2022-05-12 16:25:55 -04:00
led Black preview (#17217) 2022-05-12 16:25:55 -04:00
longformer Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
luke Black preview (#17217) 2022-05-12 16:25:55 -04:00
lxmert Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
m2m_100 Black preview (#17217) 2022-05-12 16:25:55 -04:00
marian Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
maskformer Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
mbart Black preview (#17217) 2022-05-12 16:25:55 -04:00
mbart50 Black preview (#17217) 2022-05-12 16:25:55 -04:00
megatron_bert Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
megatron_gpt2 Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
mluke Black preview (#17217) 2022-05-12 16:25:55 -04:00
mobilebert Black preview (#17217) 2022-05-12 16:25:55 -04:00
mpnet Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
mt5 Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
nystromformer Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
openai Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
opt correct opt (#17301) 2022-05-17 15:48:23 +02:00
pegasus Black preview (#17217) 2022-05-12 16:25:55 -04:00
perceiver Black preview (#17217) 2022-05-12 16:25:55 -04:00
phobert Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
plbart Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
poolformer Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
prophetnet Black preview (#17217) 2022-05-12 16:25:55 -04:00
qdqbert Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
rag Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
realm Black preview (#17217) 2022-05-12 16:25:55 -04:00
reformer Black preview (#17217) 2022-05-12 16:25:55 -04:00
regnet Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
rembert Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
resnet Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
retribert fix retribert's test_torch_encode_plus_sent_to_model (#17231) 2022-05-17 14:33:13 +02:00
roberta Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
roformer Skip RoFormer ONNX test if rjieba not installed (#16981) 2022-05-04 10:04:10 +02:00
segformer Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
sew Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
sew_d Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
speech_encoder_decoder Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
speech_to_text Black preview (#17217) 2022-05-12 16:25:55 -04:00
speech_to_text_2 Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
splinter Add support for pretraining recurring span selection to Splinter (#17247) 2022-05-17 23:42:14 +02:00
squeezebert Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
swin Add Tensorflow Swin model (#16988) 2022-05-16 22:19:53 +01:00
t5 Black preview (#17217) 2022-05-12 16:25:55 -04:00
tapas Black preview (#17217) 2022-05-12 16:25:55 -04:00
tapex Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
trajectory_transformer Add trajectory transformer (#17141) 2022-05-17 19:07:43 -04:00
transfo_xl Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
trocr Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
unispeech Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
unispeech_sat Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
van Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
vilt Black preview (#17217) 2022-05-12 16:25:55 -04:00
vision_encoder_decoder Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
vision_text_dual_encoder Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
visual_bert Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
vit ViT and Swin symbolic tracing with torch.fx (#17182) 2022-05-12 10:42:27 +02:00
vit_mae Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
wav2vec2 Black preview (#17217) 2022-05-12 16:25:55 -04:00
wav2vec2_conformer [Test] Fix W2V-Conformer integration test (#17303) 2022-05-17 18:20:36 +02:00
wav2vec2_phoneme Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
wav2vec2_with_lm Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
wavlm Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
xglm Black preview (#17217) 2022-05-12 16:25:55 -04:00
xlm Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
xlm_prophetnet Black preview (#17217) 2022-05-12 16:25:55 -04:00
xlm_roberta Black preview (#17217) 2022-05-12 16:25:55 -04:00
xlm_roberta_xl Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
xlnet Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
yolos Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
yoso Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
__init__.py Move test model folders (#17034) 2022-05-03 14:42:02 +02:00