Fix on style

intel · Jan 23, 2025 · 77a35e0 · 77a35e0
1 parent 927147c
commit 77a35e0
Showing 1 changed file with 2 additions and 2 deletions.
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
@@ -473,8 +473,8 @@ def layer_norm(self, hidden_states, layernorm_weight):
         eps = self.constant(self.rms_norm_eps)
         hidden_states = self.eltwise_div(hidden_states, self.sqrt(self.eltwise_add(variance, eps)))
         if os.environ.get("IPEX_LLM_NPU_DRIVER_VERSION", None) in ["5716", "5733"] or \
-            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_MTL", "0") == "1" or \
+           os.environ.get("IPEX_LLM_NPU_ARL", "0") == "1":
             # to support special drivers
             hidden_states = self.convert_to_fp16(hidden_states)
         else: