Update modeling_paragram_sp.py

modeling_paragram_sp.py

@@ -10,7 +10,7 @@ class ParagramSPModel(BertPreTrainedModel):
         # Initialize weights and apply final processing
         self.post_init()
 
-    def filter_input_ids(input_ids):
+    def filter_input_ids(self, input_ids):
       output = []
       len = input_ids.shape[1]
       for ids in input_ids.shape[0]:
@@ -26,7 +26,7 @@ class ParagramSPModel(BertPreTrainedModel):
     def forward(self, input_ids, attention_mask):
         print(input_ids)
         print(attention_mask)
-        input_ids = filter_input_ids(input_ids)
+        input_ids = self.filter_input_ids(input_ids)
         attention_mask = input_ids > 0
         embeddings = self.word_embeddings(input_ids)
         masked_embeddings = embeddings * attention_mask[:, :, None]