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

[NPU] fix npu save (#12614)

Browse source 
* fix npu save

* update
This commit is contained in:
Ruonan Wang 2024-12-25 17:21:16 -08:00 committed by GitHub
parent 0477fe6480
commit 9e895f04ec
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
python/llm/src/ipex_llm/transformers/npu_pipeline_model/common.py
View file

@ -27,7 +27,7 @@ def update_names_of_IR_and_export_blob(model, model_name, dir, compile_blob=True
npu_dpu_groups=None): npu_dpu_groups=None):
xml_path = os.path.join(dir, model_name + ".xml") xml_path = os.path.join(dir, model_name + ".xml")
bin_path = os.path.join(dir, model_name + ".bin") bin_path = os.path.join(dir, model_name + ".bin")
model.save(xml_path) model.serialize(xml_path, bin_path)
new_ir_path = os.path.join(dir, model_name + "_new.xml") new_ir_path = os.path.join(dir, model_name + "_new.xml")
new_bin_path = os.path.join(dir, model_name + "_new.bin") new_bin_path = os.path.join(dir, model_name + "_new.bin")
blob_path = os.path.join(dir, model_name + ".blob") blob_path = os.path.join(dir, model_name + ".blob")