From b46ac52a2795f1ea1214a5426a6fc2d958f99780 Mon Sep 17 00:00:00 2001 From: rnwang04 Date: Thu, 5 Dec 2024 16:48:36 +0800 Subject: [PATCH] fix style --- python/llm/src/ipex_llm/transformers/npu_models/convert_mp.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/python/llm/src/ipex_llm/transformers/npu_models/convert_mp.py b/python/llm/src/ipex_llm/transformers/npu_models/convert_mp.py index 12a4da5b512..71986352753 100644 --- a/python/llm/src/ipex_llm/transformers/npu_models/convert_mp.py +++ b/python/llm/src/ipex_llm/transformers/npu_models/convert_mp.py @@ -184,7 +184,8 @@ def optimize_llm_pre(model: torch.nn.Module, qtype, mixed_precision, new_lm_head = SlicedLMHead(model.lm_head.weight, split_num=split_num, bias=model.lm_head.bias, use_split=True, group_size=quantization_group_size, - asym=(qtype == "asym_int4_rtn") and (not mixed_precision)) + asym=(qtype == "asym_int4_rtn") and + (not mixed_precision)) del model.lm_head model.lm_head = new_lm_head