Add help message for no arguments provided in finetune.py and serve.py (

#169)

Signed-off-by: Wu, Xiaochang <[email protected]>

llm_on_ray/finetune/finetune.py

@@ -2,6 +2,7 @@
 
 import os
 import argparse
+import sys
 from typing import Any, Dict, Union, Optional
 
 import torch
@@ -271,6 +272,12 @@ def get_finetune_config():
         default=None,
         help="The name of the dataset to use (via the datasets library).",
     )
+
+    # Print help if no arguments were provided
+    if len(sys.argv) == 1:
+        parser.print_help(sys.stderr)
+        sys.exit(1)
+
     args = parser.parse_args()
     config_file = args.config_file
 

llm_on_ray/inference/serve.py

@@ -160,6 +160,11 @@ def main(argv=None):
         "--vllm_max_num_seqs", default=256, type=int, help="The batch size in VLLM."
     )
 
+    # Print help if no arguments were provided
+    if len(sys.argv) == 1:
+        parser.print_help(sys.stderr)
+        sys.exit(1)
+
     args = parser.parse_args(argv)
 
     ray.init(address="auto")