ImportError: cannot import name 'RobertaForRegression' from 'transformers' (/opt/anaconda3/envs/dl/lib/python3.9/site-packages/transformers/__init__.py)

#4
by renhong0322 - opened

from transformers import AutoTokenizer, RobertaModel

tokenizer = AutoTokenizer.from_pretrained("DeepChem/ChemBERTa-10M-MTR")
model = RobertaModel.from_pretrained("DeepChem/ChemBERTa-10M-MTR")

Sign up or log in to comment