Update modeling.py

modeling.py

@@ -40,15 +40,15 @@ class RobertaEmoPillars(PreTrainedModel):
                                      num_classes=config.num_labels)
         self.post_init()  # ensures HF weights init
 
-    def forward(self, input_ids=None, attention_mask=None):
-        inputs = {"input_ids": input_ids, "attention_mask": attention_mask}
-        emb = self.encoder(inputs)
-        logits = self.classifier(emb)
-        return logits
+    def forward(self, input_ids=None, attention_mask=None, **kwargs):
+    inputs = {
+        "input_ids": input_ids,
+        "attention_mask": attention_mask,
+        **kwargs
+    }
 
 
 # --- Registration for AutoModel ---
-
 try:
     AutoModel.register(AutoConfig, RobertaEmoPillars)
 except: