README.md

June 9, 2022 ยท View on GitHub

Numerical Folder:

Supporting Models: Transformer, Informer, Autoformer with different temporal embeddings

HuggingFace_EncDec Folder:

Supporting Models: BERT, Electra, RoBERTa

HuggingFace_Generation Folder:

Supporting Models: Bart, Blenderbot, LED, Pegasus, ProphetNet, Bigbird, T5

The Pre-trained Models for HuggingFace Language Models can be accessed as follows:

Model nameHuggingFace Model
T5t5-base
Bartfacebook/bart-base
Blenderbotfacebook/blenderbot-small-90M
LEDallenai/led-base-16384
Pegasusgoogle/pegasus-xsum
ProphetNetmicrosoft/prophetnet-large-uncased
Bigbirdgoogle/bigbird-pegasus-large-arxiv
Electragoogle/electra-base-generator
BERTbert-base-uncased
RoBERTaroberta-base

Core Packages:

transformers == 4.17.0
torch == 1.10.2