From 3fee838b1458134a181825f2b1018c143275f7dd Mon Sep 17 00:00:00 2001
From: binbin Deng <108676127+plusbang@users.noreply.github.com>
Date: Mon, 10 Feb 2025 11:17:58 +0800
Subject: [PATCH] [NPU] Fix of c++ convert example (#12797)

---
 .../NPU/HF-Transformers-AutoModels/LLM/CPP_Examples/convert.py  | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/python/llm/example/NPU/HF-Transformers-AutoModels/LLM/CPP_Examples/convert.py b/python/llm/example/NPU/HF-Transformers-AutoModels/LLM/CPP_Examples/convert.py
index 945100b0..d04d0448 100644
--- a/python/llm/example/NPU/HF-Transformers-AutoModels/LLM/CPP_Examples/convert.py
+++ b/python/llm/example/NPU/HF-Transformers-AutoModels/LLM/CPP_Examples/convert.py
@@ -57,6 +57,8 @@ if __name__ == "__main__":
     args = parser.parse_args()
     model_path = args.repo_id_or_model_path
     save_dir = args.save_directory
+    if not os.path.exists(save_dir):
+        os.mkdir(save_dir)
 
     tokenizer = AutoTokenizer.from_pretrained(model_path, trust_remote_code=True)