Update convert-pth-to-ggml.py

Co-authored-by: Jed Fox <git@jedfox.com>
2 years ago · 5ed9219221
parent bb40590714
commit 5ed9219221
1 changed files with 1 additions and 5 deletions
--- a/convert-pth-to-ggml.py
+++ b/convert-pth-to-ggml.py
@ -42,11 +42,7 @@ ARG_PARSER.add_argument("--output",
                        help="Model to write")
 ARGS = ARG_PARSER.parse_args()

-FTYPE_IDX = -1
-if ARGS.ftype == "f16":
-    FTYPE_IDX = 1
-elif ARGS.ftype == "f32":
-    FTYPE_IDX = 0
+FTYPE_IDX = {"f16": 1, "f32": 0}[ARGS.ftype]

 fname_hparams   = ARGS.model + "/params.json"
 fname_tokenizer = ARGS.model + "/../tokenizer.model"