From cc76ec809ab17c4e6cf5b46c22d565b36b4b850b Mon Sep 17 00:00:00 2001 From: Zhao Changmin Date: Tue, 27 Jun 2023 21:28:39 +0800 Subject: [PATCH] check out dir (#8395) --- python/llm/src/bigdl/llm/convert_model.py | 1 + 1 file changed, 1 insertion(+) 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,