# Load model directly
from transformers import AutoTokenizer, AutoModelForMaskedLM
tokenizer = AutoTokenizer.from_pretrained("SchwallerGroup/CheMatE-v0-MLM")
model = AutoModelForMaskedLM.from_pretrained("SchwallerGroup/CheMatE-v0-MLM")Quick Links
CheMatE (MLM base)
The masked-language-modeling base for
SchwallerGroup/CheMatE-v0: a ModernBERT
model pretrained on SMILES-augmented chemistry text. For embeddings, use CheMatE-v0.
Usage
from transformers import pipeline
fill = pipeline("fill-mask", model="SchwallerGroup/CheMatE-v0-MLM")
fill("Aspirin is a common [MASK].")
License
MIT.
- Downloads last month
- 21
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("fill-mask", model="SchwallerGroup/CheMatE-v0-MLM")