From 1eaaa8598074d1432040ae94379ef8c75673b3be Mon Sep 17 00:00:00 2001 From: hoshi-hiyouga Date: Thu, 21 Nov 2024 22:30:23 +0800 Subject: [PATCH] do not split save_cmd ret value Former-commit-id: d20b97e7e9514e68c33d0428512fd21b548c1131 --- src/llamafactory/webui/runner.py | 7 +------ 1 file changed, 1 insertion(+), 6 deletions(-) diff --git a/src/llamafactory/webui/runner.py b/src/llamafactory/webui/runner.py index ec549cc4..e32035a3 100644 --- a/src/llamafactory/webui/runner.py +++ b/src/llamafactory/webui/runner.py @@ -320,12 +320,7 @@ class Runner: if args.get("deepspeed", None) is not None: env["FORCE_TORCHRUN"] = "1" - cmd = [ - "llamafactory-cli", - "train", - *save_cmd(args).split(), - ] - self.trainer = Popen(cmd, env=env) + self.trainer = Popen(["llamafactory-cli", "train", save_cmd(args)], env=env) yield from self.monitor() def _form_config_dict(self, data: Dict["Component", Any]) -> Dict[str, Any]: