Update modeling_sim.py

modeling_sim.py

@@ -2,7 +2,7 @@ import torch
 from torch import nn
 from transformers import PreTrainedModel, MobileBertModel
 
-class SimModel(PreTrainedModel):
+class SimModel(MobileBertModel):
     def __init__(self, config):
         super().__init__(config)
         self.config = config