From 504b3a06c46f0bba8ea2dd20332669643a8a5220 Mon Sep 17 00:00:00 2001 From: rnwang04 Date: Wed, 25 Dec 2024 19:27:32 +0800 Subject: [PATCH] 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 49b022a36f4..687fb7aa755 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, False) + 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")