From 89a0d10c18faf6d688af2355ace1d4dcc12a77a8 Mon Sep 17 00:00:00 2001 From: Biao Wang <92444005+Wangbiao2@users.noreply.github.com> Date: Mon, 19 May 2025 17:59:39 +0800 Subject: [PATCH] [misc] fix cli (#8095) Co-authored-by: wangbiao11 --- src/llamafactory/cli.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/llamafactory/cli.py b/src/llamafactory/cli.py index f9c32d43..6eb61702 100644 --- a/src/llamafactory/cli.py +++ b/src/llamafactory/cli.py @@ -73,7 +73,7 @@ def main(): "help": partial(print, USAGE), } - command = sys.argv.pop(1) if len(sys.argv) >= 1 else "help" + command = sys.argv.pop(1) if len(sys.argv) > 1 else "help" if command == "train" and (is_env_enabled("FORCE_TORCHRUN") or (get_device_count() > 1 and not use_ray())): # launch distributed training nnodes = os.getenv("NNODES", "1")