[NPU] Fix of c++ convert example (#12797)

intel · Feb 10, 2025 · 3fee838 · 3fee838
1 parent 468d3f2
commit 3fee838
Showing 1 changed file with 2 additions and 0 deletions.
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
@@ -57,6 +57,8 @@
     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)