Update modeling_sim.py

modeling_sim.py

@@ -13,4 +13,4 @@ class SimModel(MobileBertModel):
     def forward(self, input_ids, attention_mask, token_type_ids, return_dict):
         print(input_ids, attention_mask, token_type_ids)
         print(return_dict)
-        return self.encoder(input_ids, attention_mask, token_type_ids).last_hidden_state
+        return self.encoder(input_ids=input_ids, attention_mask=attention_mask, token_type_ids=token_type_ids, return_dict=return_dict).last_hidden_state