ayo/ipex-llm
1
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 1

Fix abnormal output for Qwen2-7B when sym_int8 (#12446)

Browse source
This commit is contained in:
Yuwen Hu 2024-11-26 15:53:04 +08:00 committed by GitHub
parent 71e1f11aa6
commit 303b104c10
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 5 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
python/llm/src/ipex_llm/transformers/npu_models/convert_mp.py
View file

@ -128,6 +128,10 @@ def optimize_llm_pre(model: torch.nn.Module, qtype, mixed_precision,
from ipex_llm.transformers.npu_models.common import split_linears from ipex_llm.transformers.npu_models.common import split_linears
if quantization_group_size == 0: if quantization_group_size == 0:
n_splits_linear = 1 n_splits_linear = 1
if qtype == "sym_int8_rtn":
# do not split mlp down_proj for Qwen2-7B & sym_int8
n_splits_down_proj = 1
else:
n_splits_down_proj = 2 if model.config.intermediate_size == 18944 else 1 n_splits_down_proj = 2 if model.config.intermediate_size == 18944 else 1
else: else:
invalidInputError( invalidInputError(