diff --git a/llms/mlx_lm/lora.py b/llms/mlx_lm/lora.py index 580e3d3c..d24ff489 100644 --- a/llms/mlx_lm/lora.py +++ b/llms/mlx_lm/lora.py @@ -79,7 +79,7 @@ def build_parser(): parser.add_argument( "--data", type=str, - help="Directory with {train, valid, test}.jsonl files", + help="Directory with {train, valid, test}.jsonl files or the name of a Huggingface dataset (e.g., 'mlx-community/wikisql')", ) parser.add_argument( "--lora-layers",