diff --git a/src/llamafactory/cli.py b/src/llamafactory/cli.py
index a2ad43c4..94cc9fce 100644
--- a/src/llamafactory/cli.py
+++ b/src/llamafactory/cli.py
@@ -120,3 +120,6 @@ def main():
         print(USAGE)
     else:
         raise NotImplementedError(f"Unknown command: {command}.")
+
+if __name__ == "__main__":
+    main()