jinaai
/

xlm-roberta-flash-implementation

koukandre commited on Apr 17

Commit

f0925f9

•

1 Parent(s): 17e3855

Update modeling_xlm_roberta_for_glue.py

Files changed (1) hide show

modeling_xlm_roberta_for_glue.py CHANGED Viewed

@@ -9,7 +9,7 @@ from .modeling_bert import XLMRobertaPreTrainedModel, XLMRobertaModel
 from .configuration_bert import JinaBertConfig
-class BertForSequenceClassification(XLMRobertaPreTrainedModel):
     def __init__(self, config: JinaBertConfig):
         super().__init__(config)
         self.num_labels = config.num_labels

 from .configuration_bert import JinaBertConfig
+class XLMRobertaForSequenceClassification(XLMRobertaPreTrainedModel):
     def __init__(self, config: JinaBertConfig):
         super().__init__(config)
         self.num_labels = config.num_labels