output_dir join 'save_model'

uopensail · Nov 22, 2023 · c6cce93 · c6cce93
1 parent 6e67baf
commit c6cce93
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/damo_embedding.py b/damo_embedding.py
@@ -253,6 +253,7 @@ def save_model_for_training(model: torch.nn.Module, output_dir: str):
         output_dir (str): output dir
     """
     # remove output dir
+    output_dir = os.path.join(output_dir,"save_model")
     if os.path.exists(output_dir):
         shutil.rmtree(output_dir)
     os.makedirs(output_dir)

diff --git a/setup.py b/setup.py
@@ -157,7 +157,7 @@ def build_extension(self, ext: CMakeExtension) -> None:
 
 setup(
     name="damo-embedding",
-    version="1.1.2",
+    version="1.1.3",
     description="Python wrapper for damo, a set of fast and robust hash functions.",
     license="License :: AGLP3",
     author="timepi",