seyonec

/

PubChem10M_SMILES_BPE_60k

like 1

# Use a pipeline as a high-level helper
from transformers import pipeline

pipe = pipeline("fill-mask", model="seyonec/PubChem10M_SMILES_BPE_60k")

# Load model directly
from transformers import AutoTokenizer, AutoModelForMaskedLM

tokenizer = AutoTokenizer.from_pretrained("seyonec/PubChem10M_SMILES_BPE_60k")
model = AutoModelForMaskedLM.from_pretrained("seyonec/PubChem10M_SMILES_BPE_60k", device_map="auto")