check out dir (#8395)

2023-06-27 21:28:39 +08:00 · 2023-06-27 21:28:39 +08:00 · cc76ec809a
commit cc76ec809a
parent 4be784a49d
1 changed files with 1 additions and 0 deletions
--- 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,