From b852c895cfbc6fb1ae730adac411da4cd4934b72 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: 1e312072fb4a9f472e2d3fa7e6b4fb0aec00b566 --- 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]: