From 9e895f04ecbed45858e87a76323e513b1d231005 Mon Sep 17 00:00:00 2001 From: Ruonan Wang Date: Wed, 25 Dec 2024 17:21:16 -0800 Subject: [PATCH] [NPU] fix npu save (#12614) * fix npu save * update --- .../llm/src/ipex_llm/transformers/npu_pipeline_model/common.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/llm/src/ipex_llm/transformers/npu_pipeline_model/common.py b/python/llm/src/ipex_llm/transformers/npu_pipeline_model/common.py index 87459a99..687fb7aa 100644 --- a/python/llm/src/ipex_llm/transformers/npu_pipeline_model/common.py +++ b/python/llm/src/ipex_llm/transformers/npu_pipeline_model/common.py @@ -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")