diff --git a/src/llamafactory/cli.py b/src/llamafactory/cli.py index c7f136b3..e4846780 100644 --- 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) + command = sys.argv.pop(1) if len(sys.argv) != 1 else Command.HELP if command == Command.API: run_api() elif command == Command.CHAT: