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

store_true defaults to False, not None

Browse source
This commit is contained in:
Galunid 2023-11-07 07:25:43 +01:00
parent 05fb6f4e8c
commit 7a3433b4b6
1 changed files with 1 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

6
convert-hf-to-gguf.py
View file

@ -870,16 +870,12 @@ else:
# output in the same directory as the model by default # output in the same directory as the model by default
fname_out = dir_model / f'ggml-model-{ftype_str[ftype]}.gguf' fname_out = dir_model / f'ggml-model-{ftype_str[ftype]}.gguf'
is_big_endian = False
if args.bigendian is not None:
is_big_endian = True
print(f"Loading model: {dir_model.name}") print(f"Loading model: {dir_model.name}")
hparams = Model.load_hparams(dir_model) hparams = Model.load_hparams(dir_model)
model_class = Model.from_model_architecture(hparams["architectures"][0]) model_class = Model.from_model_architecture(hparams["architectures"][0])
model_instance = model_class(dir_model, ftype, fname_out, is_big_endian) model_instance = model_class(dir_model, ftype, fname_out, args.bigendian)
print("Set model parameters") print("Set model parameters")
model_instance.set_gguf_parameters() model_instance.set_gguf_parameters()