modules/SwissArmyTransformer/sat/model/official/roberta_model.py (5 lines of code) (raw):

from sat.model.official.bert_model import BertModel class RobertaModel(BertModel): def __init__(self, args, transformer=None, **kwargs): super(RobertaModel, self).__init__(args, transformer=transformer, **kwargs) self.del_mixin("bert-type")