style fix

intel-analytics · Dec 17, 2024 · 84f35e3 · 84f35e3
1 parent 4c5eddd
commit 84f35e3
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/python/llm/src/ipex_llm/transformers/npu_models/llama_mp.py b/python/llm/src/ipex_llm/transformers/npu_models/llama_mp.py
@@ -855,7 +855,8 @@ def run_prefill(
                         weights.append((torch.stack(l_weights, axis=0), torch.stack(scales, axis=0),
                                         torch.stack(zeros, axis=0)))
                     else:
-                        weights.append((torch.stack(l_weights, axis=0), torch.stack(scales, axis=0)))
+                        weights.append((torch.stack(l_weights, axis=0),
+                                        torch.stack(scales, axis=0)))
 
                 if hasattr(curr_layer.self_attn.rotary_emb, "cos_cached"):
                     cached_cos = curr_layer.self_attn.rotary_emb.cos_cached.to(torch.float16)