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

py : fix hparams parsing (if-else blocks)

Browse source 
ggml-ci
This commit is contained in:
Georgi Gerganov 2023-07-23 12:26:21 +03:00
parent 3fdc00f596
commit 2dac31b358
1 changed files with 5 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

10
convert.py
View file

@ -219,12 +219,12 @@ class Params:
@staticmethod
def load(model_plus: 'ModelPlus') -> 'Params':
hf_transformer_config_path = model_plus.paths[0].parent / "config.json"
orig_config_path = model_plus.paths[0].parent / "params.json"
hf_config_path = model_plus.paths[0].parent / "config.json"
orig_config_path = model_plus.paths[0].parent / "params.json"
if hf_transformer_config_path.exists():
params = Params.loadHFTransformerJson(model_plus.model, hf_transformer_config_path)
if orig_config_path.exists():
if hf_config_path.exists():
params = Params.loadHFTransformerJson(model_plus.model, hf_config_path)
else if orig_config_path.exists():
params = Params.loadOriginalParamsJson(model_plus.model, orig_config_path)
else:
params = Params.guessed(model_plus.model)