ayo/ipex-llm
1
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 1

LLM: Fix ggml return value (#8358)

Browse source 
* ggml return original value
This commit is contained in:
Zhao Changmin 2023-06-19 17:02:56 +08:00 committed by GitHub
parent d4027d7164
commit c256cd136b
1 changed files with 7 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
python/llm/src/bigdl/llm/convert_model.py
View file

@ -41,13 +41,12 @@ def llm_convert(model,
check, _used_args = _special_kwarg_check(kwargs=kwargs,
check_args=["tmp_path"])
invalidInputError(check, f"Invaid input kwargs found: {_used_args}")
ggml_convert_model(input_path=model,
output_path=outfile,
model_family=model_family,
dtype=outtype,
**_used_args,
)
return outfile
return ggml_convert_model(input_path=model,
output_path=outfile,
model_family=model_family,
dtype=outtype,
**_used_args,
)
elif model_format == "gptq":
invalidInputError(model.endswith(".pt"), "only support pytorch's .pt format now.")
invalidInputError(model_family == "llama" and outtype == 'int4',
@ -68,6 +67,7 @@ def llm_convert(model,
else:
invalidInputError(False, f"Unsupported input model_type: {model_format}")
return None
def main():
parser = argparse.ArgumentParser(description='Model Convert Parameters')