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():
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(
"-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'."
"-v", "--verbose", action="store_true",
help="Increase output verbosity."
)
parser.add_argument(
"-m", "--model-path", default="models/",
@ -34,7 +32,7 @@ def main():
)
parser.add_argument(
"--vocab-type",
const="BPE", nargs="?", choices=["BPE", "SPM"],
const="BPE", nargs="?", choices=["SPM", "BPE", "WPM"],
help="The type of vocab. Default is 'BPE'."
)
args = parser.parse_args()