Update moep.py

moep.py

@@ -294,6 +294,7 @@ class MoEPLMConfig(PretrainedConfig):
 
 class MoEPModel(PreTrainedModel):
     config_class = MoEPLMConfig
+    base_model_prefix = "model"
 
     def __init__(self, config: MoEPLMConfig):
         super().__init__(config)
@@ -443,6 +444,7 @@ class MoEPModel(PreTrainedModel):
 
 class MoEPForCausalLM(PreTrainedModel):
     config_class = MoEPLMConfig
+    base_model_prefix = "model"
 
     def __init__(self, config: MoEPLMConfig):
         super().__init__(config)