more nits

2025-09-01 12:49:50 +08:00 · 2023-12-15 10:06:14 -08:00
parent fa51553f09
commit d108c558fc
3 changed files with 27 additions and 25 deletions
--- a/lora/convert.py
+++ b/lora/convert.py
@@ -14,13 +14,13 @@ if __name__ == "__main__":
        description="Convert Mistral or Llama models to MLX.",
    )
    parser.add_argument(
-        "--torch_model",
+        "--torch-model",
        type=str,
        default="mistral-7B-v0.1/",
        help="The torch model directory",
    )
    parser.add_argument(
-        "--mlx_model",
+        "--mlx-model",
        type=str,
        default="mlx-mistral-7B-v0.1/",
        help="The directory to store the mlx model",