diff --git a/python/llm/src/bigdl/llm/convert_model.py b/python/llm/src/bigdl/llm/convert_model.py index d84ea2c5..fc195309 100644 --- a/python/llm/src/bigdl/llm/convert_model.py +++ b/python/llm/src/bigdl/llm/convert_model.py @@ -89,6 +89,7 @@ def llm_convert(model, invalidInputError(model_family == "llama" and outtype == 'int4', "Convert GPTQ models should always " "specify `--model-family llama --dtype int4` in the command line.") + os.makedirs(outfile, exist_ok=True) invalidInputError(os.path.isdir(outfile), "The output_path {} is not a directory".format(outfile)) _, _used_args = _special_kwarg_check(kwargs=kwargs,