Transformers How to use DeepChem/ChemBERTa-5M-MTR with Transformers:
# pip install -U transformers accelerate
# Load model directly
from transformers import AutoTokenizer, RobertaForRegression
tokenizer = AutoTokenizer.from_pretrained("DeepChem/ChemBERTa-5M-MTR")
model = RobertaForRegression.from_pretrained("DeepChem/ChemBERTa-5M-MTR", device_map="auto") 
# pip install -U transformers accelerate # Load model directly from transformers import AutoTokenizer, RobertaForRegression tokenizer = AutoTokenizer.from_pretrained("DeepChem/ChemBERTa-5M-MTR") model = RobertaForRegression.from_pretrained("DeepChem/ChemBERTa-5M-MTR", device_map="auto")