Small fix

intel · Jan 7, 2025 · a71d0b0 · a71d0b0
1 parent cc493e7
commit a71d0b0
Showing 1 changed file with 1 addition and 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
@@ -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)
+        model.generation_config.save_pretrained(save_directory)