Update structformer_as_hf.py

structformer_as_hf.py

@@ -1,45 +1,3 @@
-Hugging Face's logo
-Hugging Face
-Search models, datasets, users...
-Models
-Datasets
-Spaces
-Docs
-Solutions
-Pricing
-
-
-
-
-omarmomen
-/
-structformer_s1_final 
-
-like
-0
-Fill-Mask
-Transformers
-PyTorch
-structformer
-custom_code
-Model card
-Files and versions
-Community
-Settings
-structformer_s1_final
-/
-structformer_as_hf.py
-Omar
-upfdate
-a7e60f9
-3 months ago
-raw
-history
-blame
-edit
-delete
-No virus
-36.1 kB
 import torch
 import torch.nn as nn
 import torch.nn.functional as F
@@ -1162,4 +1120,4 @@ class StructformerModelForSequenceClassification(PreTrainedModel):
 
     
     def forward(self, input_ids, labels=None, **kwargs):
-        return self.model(input_ids, labels=labels, **kwargs)
+        return self.model(input_ids, labels=labels, **kwargs)