Upload RobertaEmotion

configuration_roberta_emotion.py

@@ -1,7 +1,8 @@
 from transformers import PretrainedConfig
 
+
 class RobertaEmotionConfig(PretrainedConfig):
     model_type = "ma2za/roberta-emotion"
 
     def __init__(self, **kwargs):
-        super().__init__(**kwargs)
+        super().__init__(**kwargs)

modeling_roberta_emotion.py

@@ -1,13 +1,11 @@
+import torch
 from torch.nn import CrossEntropyLoss
-
-import torch 
-
 from transformers import AutoModel, PreTrainedModel
-
 from transformers.modeling_outputs import SequenceClassifierOutput
 
 from .configuration_roberta_emotion import RobertaEmotionConfig
 
+
 class RobertaEmotion(PreTrainedModel):
     config_class = RobertaEmotionConfig
 
@@ -29,4 +27,4 @@ class RobertaEmotion(PreTrainedModel):
             loss_fct = CrossEntropyLoss()
             loss = loss_fct(logits.view(-1, self.num_labels), labels.view(-1))
 
-        return SequenceClassifierOutput(loss=loss, logits=logits)
+        return SequenceClassifierOutput(loss=loss, logits=logits)

pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:680e4142e5ffd42a300aa54b976193bfc499640ad8161ce83044dc273d128d80
+oid sha256:e69dd62f8db826d469c8d82f95887b24557ba057283a4994b3d8aca3c918e251
 size 498674549