Update modeling_sim.py

modeling_sim.py

@@ -1,8 +1,8 @@
 import torch
 from torch import nn
-from transformers import PreTrainedMobileBertModel, MobileBertModel
+from transformers import MobileBertPreTrainedModel, MobileBertModel
 
-class SimModel(MobileBertModel):
+class SimModel(MobileBertPreTrainedModel):
     def __init__(self, config):
         super().__init__(config)
         self.config = config