rename --model_path to --model-path (#151)

use same argument convention for mistral/mixtral as for llama convert.
This commit is contained in:
Daniel Strobusch
2023-12-21 15:28:57 +01:00
committed by GitHub
parent 3efb1cc2cc
commit 43b6522af2
5 changed files with 6 additions and 6 deletions

View File

@@ -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",