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

[LLM] add long time loading disclaimer for LLM model converting (#8279)

Browse source
This commit is contained in:
Pingchuan Ma (Henry) 2023-06-06 17:15:13 +08:00 committed by GitHub
parent 64bc123dd3
commit a3f353b939
1 changed files with 1 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
python/llm/src/bigdl/llm/ggml/convert.py
View file

@ -97,6 +97,7 @@ def _convert_to_ggml(model_path: str, outfile_dir: str,
os.makedirs(outfile_dir, exist_ok=True)
outtype = outtype.replace('p', '')
print("It may takes several minutes to load the original model, please wait...")
if model_family == 'llama':
_convert_llama(model_path, outfile_dir, outtype)
if model_family == 'gptneox':