from transformers import AutoTokenizer, AutoModelForMaskedLM
tokenizer = AutoTokenizer.from_pretrained("tkwoo/electra-small-generator")
model = AutoModelForMaskedLM.from_pretrained("tkwoo/electra-small-generator")
How to clone the model repo
git lfs install
git clone https://huggingface.co/tkwoo/electra-small-generator
# if you want to clone without large files β just their pointers
# prepend your git clone with the following env var:
GIT_LFS_SKIP_SMUDGE=1