From a04538e70b3b9c5628ae87695ac964f44952bcc9 Mon Sep 17 00:00:00 2001
From: hiyouga <467089858@qq.com>
Date: Sun, 24 Nov 2024 23:56:21 +0800
Subject: [PATCH] fix cli

Former-commit-id: 8792d78c82e83f319c584079d7ebfa2dccbb346a
---
 src/llamafactory/cli.py | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/src/llamafactory/cli.py b/src/llamafactory/cli.py
index 731d99e4..a2ad43c4 100644
--- a/src/llamafactory/cli.py
+++ b/src/llamafactory/cli.py
@@ -95,7 +95,8 @@ def main():
                 (
                     "torchrun --nnodes {nnodes} --node_rank {node_rank} --nproc_per_node {nproc_per_node} "
                     "--master_addr {master_addr} --master_port {master_port} {file_name} {args}"
-                ).format(
+                )
+                .format(
                     nnodes=os.getenv("NNODES", "1"),
                     node_rank=os.getenv("NODE_RANK", "0"),
                     nproc_per_node=os.getenv("NPROC_PER_NODE", str(get_device_count())),
@@ -103,8 +104,8 @@ def main():
                     master_port=master_port,
                     file_name=launcher.__file__,
                     args=" ".join(sys.argv[1:]),
-                ),
-                shell=True,
+                )
+                .split()
             )
             sys.exit(process.returncode)
         else: