From 858591e5e8e1fb4263cc90df6b8aec6103606a85 Mon Sep 17 00:00:00 2001 From: plusbang Date: Fri, 23 Aug 2024 17:36:19 +0800 Subject: [PATCH] fix --- python/llm/src/ipex_llm/transformers/npu_models/qwen2_mp.py | 1 + 1 file changed, 1 insertion(+) diff --git a/python/llm/src/ipex_llm/transformers/npu_models/qwen2_mp.py b/python/llm/src/ipex_llm/transformers/npu_models/qwen2_mp.py index 6bf5d35aab70..ec5e701fd4b9 100644 --- a/python/llm/src/ipex_llm/transformers/npu_models/qwen2_mp.py +++ b/python/llm/src/ipex_llm/transformers/npu_models/qwen2_mp.py @@ -39,6 +39,7 @@ from transformers.modeling_outputs import BaseModelOutputWithPast from ipex_llm.transformers.npu_models.mp_models_base import run_model from ipex_llm.transformers.npu_models.mp_models_base import LLMBaseNNFactory +from ipex_llm.transformers.npu_models.common import reshape_lm_head_input from transformers.modeling_outputs import CausalLMOutputWithPast from torch.nn import CrossEntropyLoss