Update embedding_model.py

embedding_model.py

@@ -55,7 +55,7 @@ class EmbeddingModel(models.Transformer):
 
     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)        
+        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}