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