Update embedding_model.py

embedding_model.py

@@ -57,8 +57,8 @@ class EmbeddingModel(models.Transformer):
         return {"model_name_or_path": self.model_name_or_path}
 
     def save(self, save_dir: str, **kwargs) -> None:
-        # self.auto_model.save_pretrained(save_dir, safe_serialization=True)
-        # self.tokenizer.save_pretrained(save_dir)
+        self.auto_model.save_pretrained(save_dir, safe_serialization=True)
+        self.tokenizer.save_pretrained(save_dir)
 
         with open(os.path.join(save_dir, "sentence_bert_config.json"), "w+") as f:
             json.dump(self.get_config_dict(), f, indent=4)