Skip to content

Commit 2ba85c8

Browse files
committed
py : minor
1 parent e36ecdc commit 2ba85c8

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

convert-falcon-hf-to-gguf.py

+4-4
Original file line numberDiff line numberDiff line change
@@ -55,10 +55,10 @@ def count_model_parts(dir_model: Path) -> int:
5555

5656
def parse_args() -> argparse.Namespace:
5757
parser = argparse.ArgumentParser(description="Convert a Falcon model to a GGML compatible file")
58-
parser.add_argument("--vocab-only", action="store_true", help="extract only the vocab")
59-
parser.add_argument("--outfile", type=Path, help="path to write to; default: based on input")
60-
parser.add_argument("model", type=Path, help="directory containing model file, or model file itself (*.bin)")
61-
parser.add_argument("ftype", type=int, choices=[0, 1], help="output format - use 0 for float32, 1 for float16", default = 1)
58+
parser.add_argument("--vocab-only", action="store_true", help="extract only the vocab")
59+
parser.add_argument("--outfile", type=Path, help="path to write to; default: based on input")
60+
parser.add_argument("model", type=Path, help="directory containing model file, or model file itself (*.bin)")
61+
parser.add_argument("ftype", type=int, help="output format - use 0 for float32, 1 for float16", choices=[0, 1], default = 1)
6262
return parser.parse_args()
6363

6464
args = parse_args()

0 commit comments

Comments
 (0)