From a71d0b02c7d1c3614ca70487848468175de04f88 Mon Sep 17 00:00:00 2001 From: Yuwen Hu Date: Tue, 7 Jan 2025 13:47:28 +0800 Subject: [PATCH] Small fix --- .../transformers/npu_pipeline_model/convert_pipeline.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/llm/src/ipex_llm/transformers/npu_pipeline_model/convert_pipeline.py b/python/llm/src/ipex_llm/transformers/npu_pipeline_model/convert_pipeline.py index 3361c8063df..16f8a7241d4 100644 --- a/python/llm/src/ipex_llm/transformers/npu_pipeline_model/convert_pipeline.py +++ b/python/llm/src/ipex_llm/transformers/npu_pipeline_model/convert_pipeline.py @@ -591,4 +591,4 @@ def convert_llm_for_deploy(model: torch.nn.Module, model.config.update(update_dict) model.config.save_pretrained(save_directory) if model.can_generate(): - model.generation_config.save_pretrained(save_directory) \ No newline at end of file + model.generation_config.save_pretrained(save_directory)