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")