vbatts/llama.cpp
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

refactor: Apply fixes to required arguments and fixes to options

Browse source
This commit is contained in:
teleprint-me 2024-05-23 21:36:31 -04:00
parent f62080adfa
commit ea4fc1095e
No known key found for this signature in database
GPG key ID: B0D11345E65C4D48
1 changed files with 5 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

12
gguf-py/scripts/hub-vocab.py
View file

@ -20,13 +20,11 @@ logger = logging.getLogger("gguf-gen-pre")
def main(): def main():
parser = argparse.ArgumentParser() parser = argparse.ArgumentParser()
parser.add_argument("hf_auth_token", help="A huggingface read auth token") parser.add_argument("auth_token", help="A huggingface read auth token")
parser.add_argument("model_repo", help="A huggingface model repository, e.g. org/model")
parser.add_argument( parser.add_argument(
"-v", "--verbose", action="store_true", help="Increase output verbosity." "-v", "--verbose", action="store_true",
) help="Increase output verbosity."
parser.add_argument(
"-r", "--model-repo", default="meta-llama/Llama-2-7b-hf",
help="The models repository. Default is 'meta-llama/Llama-2-7b-hf'."
) )
parser.add_argument( parser.add_argument(
"-m", "--model-path", default="models/", "-m", "--model-path", default="models/",
@ -34,7 +32,7 @@ def main():
) )
parser.add_argument( parser.add_argument(
"--vocab-type", "--vocab-type",
const="BPE", nargs="?", choices=["BPE", "SPM"], const="BPE", nargs="?", choices=["SPM", "BPE", "WPM"],
help="The type of vocab. Default is 'BPE'." help="The type of vocab. Default is 'BPE'."
) )
args = parser.parse_args() args = parser.parse_args()