Skip to content

Commit

Permalink
[NPU] fix npu save (#12614)
Browse files Browse the repository at this point in the history
* fix npu save

* update
  • Loading branch information
rnwang04 authored Dec 26, 2024
1 parent 0477fe6 commit 9e895f0
Showing 1 changed file with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -27,7 +27,7 @@ def update_names_of_IR_and_export_blob(model, model_name, dir, compile_blob=True
npu_dpu_groups=None):
xml_path = os.path.join(dir, model_name + ".xml")
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_bin_path = os.path.join(dir, model_name + "_new.bin")
blob_path = os.path.join(dir, model_name + ".blob")
Expand Down

0 comments on commit 9e895f0

Please sign in to comment.