mirror of
https://github.com/ml-explore/mlx-examples.git
synced 2025-09-01 04:14:38 +08:00
rename --model_path to --model-path (#151)
use same argument convention for mistral/mixtral as for llama convert.
This commit is contained in:
@@ -10,7 +10,7 @@ import torch
|
||||
if __name__ == "__main__":
|
||||
parser = argparse.ArgumentParser(description="Convert Mistral weights to MLX.")
|
||||
parser.add_argument(
|
||||
"--model_path",
|
||||
"--model-path",
|
||||
type=str,
|
||||
default="mistral-7B-v0.1/",
|
||||
help="The path to the Mistral model. The MLX weights will also be saved there.",
|
||||
|
@@ -225,7 +225,7 @@ def generate(prompt: mx.array, model: Mistral, temp: Optional[float] = 0.0):
|
||||
if __name__ == "__main__":
|
||||
parser = argparse.ArgumentParser(description="Mistral inference script")
|
||||
parser.add_argument(
|
||||
"--model_path",
|
||||
"--model-path",
|
||||
type=str,
|
||||
default="mistral-7B-v0.1",
|
||||
help="The path to the model weights and tokenizer",
|
||||
|
Reference in New Issue
Block a user