Upload folder using huggingface_hub

modeling_phobert_aspect.py

@@ -40,10 +40,11 @@ class PhoBERTClassifier(PreTrainedModel):
 
         self.post_init()
 
-    def forward(self, input_ids, attention_mask):
+    def forward(self, input_ids=None, attention_mask=None, token_type_ids=None,**kwargs):
         outputs = self.phobert(
             input_ids=input_ids,
-            attention_mask=attention_mask
+            attention_mask=attention_mask,
+            token_type_ids=token_type_ids
         )
 
         pooled = outputs.last_hidden_state[:, 0, :]
@@ -51,3 +52,4 @@ class PhoBERTClassifier(PreTrainedModel):
 
         logits = [clf(pooled) for clf in self.classifiers]
         return torch.stack(logits, dim=1)
+