from transformers import BertPreTrainedModel, RobertaModel import torch class RobertaForMD(BertPreTrainedModel): # Metaphor Detection, modified from BertForTokenClassification def __init__(self, config): super().__init__(config) self.num_labels = config.num_labels self…

Getting import error

prajjwal1 September 2, 2020, 8:13am 4

Oh, sorry I mistook for AutoModel. Can you just try doing this instead ?
AutoModel.from_pretrained('roberta-large', config). This should work.

Topic		Replies	Views
Robertaforquestionanswering 🤗Transformers	1	2201	August 3, 2020
ModuleNotFoundError: No module named 'transformers.modeling_roberta' 🤗Transformers	2	2363	November 16, 2021
BertPreTrainedModel and RobertaPreTrainedModel works, however PreTrainedModel does not work 🤗Transformers	0	1073	February 25, 2022
Cannot import newly uploaded model Models	1	2279	November 20, 2020
Training RoBERTa from scratch: error? 🤗Transformers	0	596	August 26, 2021