浏览代码

safe output path in webui

hiyouga 2 周之前
父节点
当前提交
3c2a992caa
共有 1 个文件被更改,包括 3 次插入2 次删除
  1. 3 2
      src/llamafactory/webui/common.py

+ 3 - 2
src/llamafactory/webui/common.py

@@ -37,8 +37,9 @@ DEFAULT_SAVE_DIR = "saves"
 USER_CONFIG = "user_config.yaml"
 
 
-def get_save_dir(*args) -> os.PathLike:
-    return os.path.join(DEFAULT_SAVE_DIR, *args)
+def get_save_dir(*paths: str) -> os.PathLike:
+    paths = (path.replace(os.path.sep, "").replace(" ", "").strip() for path in paths)
+    return os.path.join(DEFAULT_SAVE_DIR, *paths)
 
 
 def get_config_path() -> os.PathLike: