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

Make vocab_only argument optional

Browse source 
Positional arguments must be explicitly optional to use default values, otherwise the examples in the readme fail to run.
This commit is contained in:
Douglas Livingstone 2023-03-21 15:54:15 +00:00 • committed by GitHub
parent 715d292ee0
commit a8e27918b9
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
convert-pth-to-ggml.py
View file

@ -29,7 +29,7 @@ def parse_args():
parser = argparse.ArgumentParser(description='Convert a LLaMA model checkpoint to a ggml compatible file')
parser.add_argument('dir_model', help='directory containing the model checkpoint')
parser.add_argument('ftype', type=int, choices=[0, 1], default=1, help='file type (0: float32, 1: float16)')
parser.add_argument('vocab_only', type=bool, default=False, help='only write vocab to file')
parser.add_argument('vocab_only', type=bool, nargs='?', default=False, help='only write vocab to file')
return parser.parse_args()
def get_n_parts(dim):