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

LLM: fix ModuleNotFoundError when use llm-cli (#8378)

Browse source
This commit is contained in:
binbin Deng 2023-06-21 15:03:14 +08:00 committed by GitHub
parent 7296453f07
commit 03c5fb71a8
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -15,8 +15,6 @@
# #
from bigdl.llm.ggml.convert_model import convert_model as ggml_convert_model
from bigdl.llm.gptq.convert.convert_gptq_to_ggml import convert_gptq2ggml
from bigdl.llm.utils.common import invalidInputError from bigdl.llm.utils.common import invalidInputError
import argparse import argparse
import os import os
@ -77,6 +75,7 @@ def llm_convert(model,
:return: the path string to the converted lower precision checkpoint. :return: the path string to the converted lower precision checkpoint.
""" """
if model_format == "pth": if model_format == "pth":
from bigdl.llm.ggml.convert_model import convert_model as ggml_convert_model
_, _used_args = _special_kwarg_check(kwargs=kwargs, _, _used_args = _special_kwarg_check(kwargs=kwargs,
check_args=["tmp_path"]) check_args=["tmp_path"])
return ggml_convert_model(input_path=model, return ggml_convert_model(input_path=model,
@ -86,6 +85,7 @@ def llm_convert(model,
**_used_args, **_used_args,
) )
elif model_format == "gptq": elif model_format == "gptq":
from bigdl.llm.gptq.convert.convert_gptq_to_ggml import convert_gptq2ggml
invalidInputError(model_family == "llama" and outtype == 'int4', invalidInputError(model_family == "llama" and outtype == 'int4',
"Convert GPTQ models should always " "Convert GPTQ models should always "
"specify `--model-family llama --dtype int4` in the command line.") "specify `--model-family llama --dtype int4` in the command line.")