|
|
@ -42,14 +42,9 @@ The model name in string. |
|
|
|
The default model name is "bert-base-uncased". |
|
|
|
|
|
|
|
Supported model names: |
|
|
|
<details> |
|
|
|
<summary>Test</summary> |
|
|
|
|
|
|
|
* a |
|
|
|
* b |
|
|
|
</details> |
|
|
|
<details><summary>Bert</summary> |
|
|
|
|
|
|
|
- [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) |
|
|
|
</details> |
|
|
|
|
|
|
|
<details><summary>BigBird</summary> |
|
|
|
|
|
|
|
- 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) |
|
|
|
</details> |
|
|
|
|
|
|
|
<details><summary>Albert</summary> |
|
|
|
|
|
|
|
- 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) |
|
|
|
</details> |
|
|
|
|
|
|
|
<details><summary>Bart</summary> |
|
|
|
|
|
|
|
- facebook/bart-large |
|
|
|
- [BertGeneration](https://huggingface.co/docs/transformers/v4.16.2/en/model_doc/bert-generation) |
|
|
|
</details> |
|
|
|
|
|
|
|
<details><summary>BertGeneration</summary> |
|
|
|
|
|
|
|
- google/bert_for_seq_generation_L-24_bbc_encoder |
|
|
|
- [BigBirdPegasus](https://huggingface.co/docs/transformers/v4.16.2/en/model_doc/bigbird_pegasus) |
|
|
|
</details> |
|
|
|
|
|
|
|
<details><summary>BigBirdPegasus</summary> |
|
|
|
|
|
|
|
- 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) |
|
|
|
</details> |
|
|
|
|
|
|
|
<details><summary>CamemBert</summary> |
|
|
|
|
|
|
|
- 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) |
|
|
|
</details> |
|
|
|
|
|
|
|
<details><summary>Canine</summary> |
|
|
|
|
|
|
|
- google/canine-s |
|
|
|
- google/canine-c |
|
|
|
- [Convbert](https://huggingface.co/docs/transformers/v4.16.2/en/model_doc/convbert) |
|
|
|
</details> |
|
|
|
|
|
|
|
<details><summary>Convbert</summary> |
|
|
|
|
|
|
|
- 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) |
|
|
|
</details> |
|
|
|
|
|
|
|
<details><summary>CTRL</summary> |
|
|
|
- ctrl |
|
|
|
- [DeBERTa](https://huggingface.co/docs/transformers/v4.16.2/en/model_doc/deberta) |
|
|
|
</details> |
|
|
|
|
|
|
|
<details><summary>DeBERTa</summary> |
|
|
|
|
|
|
|
- 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) |
|
|
|
</details> |
|
|
|
|
|
|
|
<details><summary>DistilBert</summary> |
|
|
|
|
|
|
|
- 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) |
|
|
|
</details> |
|
|
|
|
|
|
|
<details><summary>Electral</summary> |
|
|
|
|
|
|
|
- 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) |
|
|
|
</details> |
|
|
|
|
|
|
|
<details><summary>Flaubert</summary> |
|
|
|
|
|
|
|
- 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) |
|
|
|
</details> |
|
|
|
|
|
|
|
<details><summary>FNet</summary> |
|
|
|
|
|
|
|
- google/fnet-base |
|
|
|
- google/fnet-large |
|
|
|
- [FSMT](https://huggingface.co/docs/transformers/v4.16.2/en/model_doc/fsmt) |
|
|
|
</details> |
|
|
|
|
|
|
|
<details><summary>FSMT</summary> |
|
|
|
|
|
|
|
- facebook/wmt19-ru-en |
|
|
|
- [Funnel](https://huggingface.co/docs/transformers/v4.16.2/en/model_doc/funnel) |
|
|
|
</details> |
|
|
|
|
|
|
|
<details><summary>Funnel</summary> |
|
|
|
- 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) |
|
|
|
</details> |
|
|
|
|
|
|
|
<details><summary>GPT</summary> |
|
|
|
|
|
|
|
- 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) |
|
|
|
</details> |
|
|
|
|
|
|
|
<details><summary>I-Bert</summary> |
|
|
|
|
|
|
|
- kssteven/ibert-roberta-base |
|
|
|
- [LED](https://huggingface.co/docs/transformers/v4.16.2/en/model_doc/led) |
|
|
|
</details> |
|
|
|
|
|
|
|
<details><summary>LED</summary> |
|
|
|
|
|
|
|
- allenai/led-base-16384 |
|
|
|
- [MobileBert](https://huggingface.co/docs/transformers/v4.16.2/en/model_doc/mobilebert) |
|
|
|
</details> |
|
|
|
|
|
|
|
<details><summary>MobileBert</summary> |
|
|
|
|
|
|
|
- google/mobilebert-uncased |
|
|
|
- [MPNet](https://huggingface.co/docs/transformers/v4.16.2/en/model_doc/mpnet) |
|
|
|
</details> |
|
|
|
|
|
|
|
<details><summary>MPNet</summary> |
|
|
|
|
|
|
|
- microsoft/mpnet-base |
|
|
|
- [Nystromformer](https://huggingface.co/docs/transformers/v4.16.2/en/model_doc/nystromformer) |
|
|
|
</details> |
|
|
|
|
|
|
|
<details><summary>Nystromformer</summary> |
|
|
|
|
|
|
|
- uw-madison/nystromformer-512 |
|
|
|
- [Reformer](https://huggingface.co/docs/transformers/v4.16.2/en/model_doc/reformer) |
|
|
|
</details> |
|
|
|
|
|
|
|
<details><summary>Reformer</summary> |
|
|
|
|
|
|
|
- google/reformer-crime-and-punishment |
|
|
|
- [Splinter](https://huggingface.co/docs/transformers/v4.16.2/en/model_doc/splinter) |
|
|
|
</details> |
|
|
|
|
|
|
|
<details><summary>Splinter</summary> |
|
|
|
|
|
|
|
- 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) |
|
|
|
</details> |
|
|
|
|
|
|
|
<details><summary>SqueezeBert</summary> |
|
|
|
|
|
|
|
- squeezebert/squeezebert-uncased |
|
|
|
- squeezebert/squeezebert-mnli |
|
|
|
- squeezebert/squeezebert-mnli-headless |
|
|
|
- [TransfoXL](https://huggingface.co/docs/transformers/v4.16.2/en/model_doc/transfo-xl) |
|
|
|
</details> |
|
|
|
|
|
|
|
<details><summary>TransfoXL</summary> |
|
|
|
|
|
|
|
- transfo-xl-wt103 |
|
|
|
- [XLM](https://huggingface.co/docs/transformers/v4.16.2/en/model_doc/xlm) |
|
|
|
</details> |
|
|
|
|
|
|
|
<details><summary>XLM</summary> |
|
|
|
|
|
|
|
- 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) |
|
|
|
</details> |
|
|
|
|
|
|
|
<details><summary>XLMRoberta</summary> |
|
|
|
|
|
|
|
- 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) |
|
|
|
</details> |
|
|
|
|
|
|
|
<details><summary>XLNet</summary> |
|
|
|
|
|
|
|
- xlnet-base-cased |
|
|
|
- xlnet-large-cased |
|
|
|
- [Yoso](https://huggingface.co/docs/transformers/v4.16.2/en/model_doc/yoso) |
|
|
|
- uw-madison/yoso-4096 |
|
|
|
</details> |
|
|
|
|
|
|
|
<details><summary>Yoso</summary> |
|
|
|
|
|
|
|
- uw-madison/yoso-4096 |
|
|
|
</details> |
|
|
|
|
|
|
|
<br /> |
|
|
|
|
|
|
|