update data argument help

2025-09-01 12:49:50 +08:00 · 2024-09-29 14:44:12 +08:00
parent a83ca489a4
commit 6d46cbb4a6
1 changed files with 1 additions and 1 deletions
--- 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",