Update modeling_sim.py

modeling_sim.py

@@ -1,12 +1,12 @@
 import torch
 from torch import nn
-from transformers import MobileBert
+from transformers import MobileBertModel
 
 class SimModel(PreTrainedModel):
     def __init__(self, config):
         super().__init__(config)
         self.config = config
-        self.encoder = MobileBert(config)
+        self.encoder = MobileBertModel(config)
         # Initialize weights and apply final processing
         self.post_init()