Getting import error

This worked for me

model = RobertaForMD.from_pretrained("roberta-large")