Update modeling.py

modeling.py

@@ -120,6 +120,21 @@ class MARModel(PreTrainedModel):
     #     state_dict = torch.load(safetensors_path, map_location='cpu')
     #     model.model.load_state_dict(state_dict)
     #     return model
+    @classmethod
+    def from_pretrained(cls, pretrained_model_name_or_path, *model_args, **kwargs):
+        config = MARConfig.from_pretrained(pretrained_model_name_or_path, *model_args, **kwargs)
+        
+        model = cls(config)
+        
+        safetensors_path = hf_hub_download(
+            repo_id=pretrained_model_name_or_path, 
+            filename="model.safetensors"
+        )
+        state_dict = load_file(safetensors_path)
+        
+        model.model.load_state_dict(state_dict)
+        
+        return model
 
 
     def save_pretrained(self, save_directory):