From 1acf94dea553c1f38818406f59168bbcd996ece5 Mon Sep 17 00:00:00 2001 From: plusbang Date: Tue, 24 Dec 2024 15:07:30 +0800 Subject: [PATCH] fix code style --- .../llm/src/ipex_llm/transformers/npu_models/mp_models_base.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/llm/src/ipex_llm/transformers/npu_models/mp_models_base.py b/python/llm/src/ipex_llm/transformers/npu_models/mp_models_base.py index 9df88cd3ab1..39d5888e230 100644 --- a/python/llm/src/ipex_llm/transformers/npu_models/mp_models_base.py +++ b/python/llm/src/ipex_llm/transformers/npu_models/mp_models_base.py @@ -494,7 +494,7 @@ def apply_rotary_pos_emb(self, *, q, k, cos, sin, position_ids, num_heads, seq_len, head_dim): if position_ids is not None: if os.environ.get("IPEX_LLM_NPU_MTL", "0") == "1" or\ - os.environ.get("IPEX_LLM_NPU_ARL", "0") == "1": + os.environ.get("IPEX_LLM_NPU_ARL", "0") == "1": position_ids = self.reshape(position_ids, [-1]) else: position_ids = self.squeeze(position_ids)