]> git.djapps.eu Git - pkg/ggml/sources/llama.cpp/commitdiff
py : minor
authorGeorgi Gerganov <redacted>
Mon, 4 Sep 2023 19:50:50 +0000 (22:50 +0300)
committerGeorgi Gerganov <redacted>
Mon, 4 Sep 2023 19:50:50 +0000 (22:50 +0300)
convert-falcon-hf-to-gguf.py

index 271e589728e6449b60bd7253df60758aa6716374..6ed2b88c6712cb0935bcae9b953802a2b06611cb 100755 (executable)
@@ -55,10 +55,10 @@ def count_model_parts(dir_model: Path) -> int:
 
 def parse_args() -> argparse.Namespace:
     parser = argparse.ArgumentParser(description="Convert a Falcon model to a GGML compatible file")
-    parser.add_argument("--vocab-only",  action="store_true",    help="extract only the vocab")
-    parser.add_argument("--outfile",     type=Path,              help="path to write to; default: based on input")
-    parser.add_argument("model",         type=Path,              help="directory containing model file, or model file itself (*.bin)")
-    parser.add_argument("ftype",     type=int, choices=[0, 1],   help="output format - use 0 for float32, 1 for float16", default = 1)
+    parser.add_argument("--vocab-only", action="store_true", help="extract only the vocab")
+    parser.add_argument("--outfile",    type=Path,           help="path to write to; default: based on input")
+    parser.add_argument("model",        type=Path,           help="directory containing model file, or model file itself (*.bin)")
+    parser.add_argument("ftype",        type=int,            help="output format - use 0 for float32, 1 for float16", choices=[0, 1], default = 1)
     return parser.parse_args()
 
 args = parse_args()