Update convert-pth-to-ggml.py
Co-authored-by: Jed Fox <git@jedfox.com>
This commit is contained in:
parent
bb40590714
commit
5ed9219221
1 changed files with 1 additions and 5 deletions
|
@ -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"
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue