Update modeling_metalatte.py

modeling_metalatte.py

@@ -218,17 +218,6 @@ class MultitaskProteinModel(PreTrainedModel):
 
         # Initialize weights and apply final processing
         self.post_init()
-    
-    @classmethod
-    def from_pretrained(cls, pretrained_model_name_or_path, *model_args, **kwargs):
-        config = kwargs.pop("config", None)
-        if config is None:
-            config = MetaLATTEConfig.from_pretrained(pretrained_model_name_or_path)
-        
-        model = cls(config)        
-        state_dict = torch.load(f"{pretrained_model_name_or_path}/pytorch_model.bin", map_location=torch.device('cpu'))['state_dict']
-        model.load_state_dict(state_dict, strict=False)
-        return model
         
     def forward(self, input_ids, attention_mask=None):
         outputs = self.esm_model(input_ids=input_ids, attention_mask=attention_mask, output_hidden_states=True)