From 02f716907ec532a8525f17892062825b86d0c6a9 Mon Sep 17 00:00:00 2001
From: hiyouga <467089858@qq.com>
Date: Sat, 18 May 2024 22:42:28 +0800
Subject: [PATCH] safe output path in webui

Former-commit-id: 3c2a992caa45c9121cd1ccac3de9bbcf1a9b058b
---
 src/llamafactory/webui/common.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/src/llamafactory/webui/common.py b/src/llamafactory/webui/common.py
index c63e9d74..ea82fd88 100644
--- a/src/llamafactory/webui/common.py
+++ b/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: