Update embedding_model.py

embedding_model.py

@@ -53,6 +53,10 @@ class EmbeddingModel(models.Transformer):
 
         return output
 
+    def _load_model(self, model_name_or_path, config, cache_dir, backend, **model_args) -> None:
+        """Loads the transformer model"""
+        return self.auto_model = AutoModel.from_pretrained(model_name_or_path, config=config, cache_dir=cache_dir, trust_remote_code=True, **model_args)        
+
     def get_config_dict(self) -> dict[str, str]:
         return {"model_name_or_path": self.model_name_or_path}