from transformers import AutoTokenizer, AutoModelForMaskedLM
tokenizer = AutoTokenizer.from_pretrained("codistai/codeBERT-small-v2")
model = AutoModelForMaskedLM.from_pretrained("codistai/codeBERT-small-v2")
How to clone the model repo
git lfs install
git clone https://huggingface.co/codistai/codeBERT-small-v2
# 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