Update modeling_sim.py

modeling_sim.py

@@ -10,6 +10,6 @@ class SimModel(MobileBertModel):
         # Initialize weights and apply final processing
         self.post_init()
 
-    def forward(self, input_ids):
-        print(input_ids)
-        return self.encoder(input_ids).last_hidden_state
+    def forward(self, input_ids, attention_mask):
+        print(input_ids, attention_mask)
+        return self.encoder(input_ids, attention_mask).last_hidden_state