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

2025-02-10 11:17:58 +08:00 · 2025-02-10 11:17:58 +08:00 · 3fee838b14
commit 3fee838b14
parent 468d3f22fc
1 changed files with 2 additions and 0 deletions
--- 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)