Update cli.py

Former-commit-id: 9db6126496ec9e834541823715f700f92b3968c7
2024-06-24 23:21:10 +08:00
parent 05abe47c8b
commit 8bcdb6f52c
1 changed files with 1 additions and 1 deletions
--- a/src/llamafactory/cli.py
+++ b/src/llamafactory/cli.py
@@ -74,7 +74,7 @@ class Command(str, Enum):


 def main():
-    command = sys.argv.pop(1) if len(sys.argv) > 0 else Command.HELP
+    command = sys.argv.pop(1) if len(sys.argv) != 1 else Command.HELP
    if command == Command.API:
        run_api()
    elif command == Command.CHAT: