ayo/ipex-llm
1
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 1

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

Browse source
This commit is contained in:
binbin Deng 2025-02-10 11:17:58 +08:00 committed by GitHub
parent 468d3f22fc
commit 3fee838b14
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 2 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
python/llm/example/NPU/HF-Transformers-AutoModels/LLM/CPP_Examples/convert.py
View file

@ -57,6 +57,8 @@ if __name__ == "__main__":
args = parser.parse_args() args = parser.parse_args()
model_path = args.repo_id_or_model_path model_path = args.repo_id_or_model_path
save_dir = args.save_directory 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) tokenizer = AutoTokenizer.from_pretrained(model_path, trust_remote_code=True)