Upload modeling_te3s_head.py with huggingface_hub

modeling_te3s_head.py

@@ -1,7 +1,6 @@
-from __future__ import annotations
-
 import torch
 import torch.nn as nn
+from typing import Optional
 from transformers import PreTrainedModel, PretrainedConfig
 
 
@@ -44,8 +43,8 @@ class TextEmbedding3SmallSentimentHead(PreTrainedModel):
     def forward(
         self,
         inputs_embeds: torch.FloatTensor,
-        labels: torch.LongTensor | None = None,
-        **_: dict,
+        labels: Optional[torch.LongTensor] = None,
+        **kwargs,
     ):
         logits = self.net(inputs_embeds)
         loss = None