jinaai
/

xlm-roberta-flash-implementation

koukandre commited on Apr 17, 2024

Commit

102e7bc

verified ·

1 Parent(s): 4dd727d

Update modeling_xlm_roberta_for_glue.py

Files changed (1) hide show

modeling_xlm_roberta_for_glue.py CHANGED Viewed

@@ -5,7 +5,7 @@ from torch import nn
 from torch.nn import CrossEntropyLoss, MSELoss, BCEWithLogitsLoss
 from transformers.modeling_outputs import SequenceClassifierOutput, QuestionAnsweringModelOutput, TokenClassifierOutput
-from .modeling_bert import XLMRobertaPreTrainedModel, XLMRobertaModel
 from .configuration_xlm_roberta import XLMRobertaFlashConfig

 from torch.nn import CrossEntropyLoss, MSELoss, BCEWithLogitsLoss
 from transformers.modeling_outputs import SequenceClassifierOutput, QuestionAnsweringModelOutput, TokenClassifierOutput
+from .modeling_xlm_roberta import XLMRobertaPreTrainedModel, XLMRobertaModel
 from .configuration_xlm_roberta import XLMRobertaFlashConfig