diff --git a/README.md b/README.md index d5c840d..ccea651 100644 --- a/README.md +++ b/README.md @@ -42,14 +42,9 @@ The model name in string. The default model name is "bert-base-uncased". Supported model names: -
-Test - * a - * b -
+
Bert -- [Bert](https://huggingface.co/docs/transformers/v4.16.2/en/model_doc/bert) - bert-base-cased - bert-large-cased - bert-base-multilingual-uncased @@ -69,11 +64,17 @@ Supported model names: - TurkuNLP/bert-base-finnish-cased-v1 - TurkuNLP/bert-base-finnish-uncased-v1 - wietsedv/bert-base-dutch-cased -- [BigBird](https://huggingface.co/docs/transformers/v4.16.2/en/model_doc/big_bird) +
+ +
BigBird + - google/bigbird-roberta-base - google/bigbird-roberta-large - google/bigbird-base-trivia-itc -- [Albert](https://huggingface.co/docs/transformers/v4.16.2/en/model_doc/albert) +
+ +
Albert + - albert-base-v1 - albert-large-v1 - albert-xlarge-v1 @@ -82,28 +83,51 @@ Supported model names: - albert-large-v2 - albert-xlarge-v2 - albert-xxlarge-v2 -- [Bart](https://huggingface.co/docs/transformers/v4.16.2/en/model_doc/bart) +
+ +
Bart + - facebook/bart-large -- [BertGeneration](https://huggingface.co/docs/transformers/v4.16.2/en/model_doc/bert-generation) +
+ +
BertGeneration + - google/bert_for_seq_generation_L-24_bbc_encoder -- [BigBirdPegasus](https://huggingface.co/docs/transformers/v4.16.2/en/model_doc/bigbird_pegasus) +
+ +
BigBirdPegasus + - google/bigbird-pegasus-large-arxiv - google/bigbird-pegasus-large-pubmed - google/bigbird-pegasus-large-bigpatent -- [CamemBert](https://huggingface.co/docs/transformers/v4.16.2/en/model_doc/camembert) +
+ +
CamemBert + - camembert-base - Musixmatch/umberto-commoncrawl-cased-v1 - Musixmatch/umberto-wikipedia-uncased-v1 -- [Canine](https://huggingface.co/docs/transformers/v4.16.2/en/model_doc/canine) +
+ +
Canine + - google/canine-s - google/canine-c -- [Convbert](https://huggingface.co/docs/transformers/v4.16.2/en/model_doc/convbert) +
+ +
Convbert + - YituTech/conv-bert-base - YituTech/conv-bert-medium-small - YituTech/conv-bert-small -- [CTRL](https://huggingface.co/docs/transformers/v4.16.2/en/model_doc/ctrl) +
+ +
CTRL - ctrl -- [DeBERTa](https://huggingface.co/docs/transformers/v4.16.2/en/model_doc/deberta) +
+ +
DeBERTa + - microsoft/deberta-base - microsoft/deberta-large - microsoft/deberta-xlarge @@ -114,7 +138,10 @@ Supported model names: - microsoft/deberta-v2-xxlarge - microsoft/deberta-v2-xlarge-mnli - microsoft/deberta-v2-xxlarge-mnli -- [DistilBert](https://huggingface.co/docs/transformers/v4.16.2/en/model_doc/distilbert) +
+ +
DistilBert + - distilbert-base-uncased - distilbert-base-uncased-distilled-squad - distilbert-base-cased @@ -122,24 +149,38 @@ Supported model names: - distilbert-base-german-cased - distilbert-base-multilingual-cased - distilbert-base-uncased-finetuned-sst-2-english -- [Electra](https://huggingface.co/docs/transformers/v4.16.2/en/model_doc/electra) +
+ +
Electral + - google/electra-small-generator - google/electra-base-generator - google/electra-large-generator - google/electra-small-discriminator - google/electra-base-discriminator - google/electra-large-discriminator -- [Flaubert](https://huggingface.co/docs/transformers/v4.16.2/en/model_doc/flaubert) +
+ +
Flaubert + - flaubert/flaubert_small_cased - flaubert/flaubert_base_uncased - flaubert/flaubert_base_cased - flaubert/flaubert_large_cased -- [FNet](https://huggingface.co/docs/transformers/v4.16.2/en/model_doc/fnet) +
+ +
FNet + - google/fnet-base - google/fnet-large -- [FSMT](https://huggingface.co/docs/transformers/v4.16.2/en/model_doc/fsmt) +
+ +
FSMT + - facebook/wmt19-ru-en -- [Funnel](https://huggingface.co/docs/transformers/v4.16.2/en/model_doc/funnel) +
+ +
Funnel - funnel-transformer/small - funnel-transformer/small-base - funnel-transformer/medium @@ -150,7 +191,10 @@ Supported model names: - funnel-transformer/large-base - funnel-transformer/xlarge-base - funnel-transformer/xlarge -- [GPT](https://huggingface.co/docs/transformers/v4.16.2/en/model_doc/openai-gpt) +
+ +
GPT + - openai-gpt - gpt2 - gpt2-medium @@ -159,30 +203,60 @@ Supported model names: - distilgpt2 - EleutherAI/gpt-neo-1.3B - EleutherAI/gpt-j-6B -- [I-Bert](https://huggingface.co/docs/transformers/v4.16.2/en/model_doc/ibert) +
+ +
I-Bert + - kssteven/ibert-roberta-base -- [LED](https://huggingface.co/docs/transformers/v4.16.2/en/model_doc/led) +
+ +
LED + - allenai/led-base-16384 -- [MobileBert](https://huggingface.co/docs/transformers/v4.16.2/en/model_doc/mobilebert) +
+ +
MobileBert + - google/mobilebert-uncased -- [MPNet](https://huggingface.co/docs/transformers/v4.16.2/en/model_doc/mpnet) +
+ +
MPNet + - microsoft/mpnet-base -- [Nystromformer](https://huggingface.co/docs/transformers/v4.16.2/en/model_doc/nystromformer) +
+ +
Nystromformer + - uw-madison/nystromformer-512 -- [Reformer](https://huggingface.co/docs/transformers/v4.16.2/en/model_doc/reformer) +
+ +
Reformer + - google/reformer-crime-and-punishment -- [Splinter](https://huggingface.co/docs/transformers/v4.16.2/en/model_doc/splinter) +
+ +
Splinter + - tau/splinter-base - tau/splinter-base-qass - tau/splinter-large - tau/splinter-large-qass -- [SqueezeBert](https://huggingface.co/docs/transformers/v4.16.2/en/model_doc/squeezebert) +
+ +
SqueezeBert + - squeezebert/squeezebert-uncased - squeezebert/squeezebert-mnli - squeezebert/squeezebert-mnli-headless -- [TransfoXL](https://huggingface.co/docs/transformers/v4.16.2/en/model_doc/transfo-xl) +
+ +
TransfoXL + - transfo-xl-wt103 -- [XLM](https://huggingface.co/docs/transformers/v4.16.2/en/model_doc/xlm) +
+ +
XLM + - xlm-mlm-en-2048 - xlm-mlm-ende-1024 - xlm-mlm-enfr-1024 @@ -193,19 +267,28 @@ Supported model names: - xlm-clm-ende-1024 - xlm-mlm-17-1280 - xlm-mlm-100-1280 -- [XLMRoberta](https://huggingface.co/docs/transformers/v4.16.2/en/model_doc/xlm-roberta) +
+ +
XLMRoberta + - xlm-roberta-base - xlm-roberta-large - xlm-roberta-large-finetuned-conll02-dutch - xlm-roberta-large-finetuned-conll02-spanish - xlm-roberta-large-finetuned-conll03-english - xlm-roberta-large-finetuned-conll03-german -- [XLNet](https://huggingface.co/docs/transformers/v4.16.2/en/model_doc/xlnet) +
+ +
XLNet + - xlnet-base-cased - xlnet-large-cased -- [Yoso](https://huggingface.co/docs/transformers/v4.16.2/en/model_doc/yoso) - - uw-madison/yoso-4096 +
+ +
Yoso + - uw-madison/yoso-4096 +