Update modeling_sim.py

modeling_sim.py

@@ -10,7 +10,7 @@ class SimModel(MobileBertPreTrainedModel):
         # Initialize weights and apply final processing
         self.post_init()
 
-    def forward(self, input_ids, attention_mask, token_type_ids, return_dict):
+    def forward(self, input_ids, attention_mask, return_dict):
         print(input_ids)
         embeddings = self.word_embeddings(input_ids)
         masked_embeddings = embeddings * attention_mask[:, :, None]