style

intel-analytics · Dec 3, 2024 · 03bf892 · 03bf892
1 parent b62ae49
commit 03bf892
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 4 deletions.
diff --git a/python/llm/src/ipex_llm/transformers/npu_models/convert.py b/python/llm/src/ipex_llm/transformers/npu_models/convert.py
@@ -445,9 +445,11 @@ def causal_lm_forward(
         input_list = input_ids[0]
     input_length = len(input_list)
     if input_length > 1:
-        logits = run_prefill_with_logits(self.model_ptr, input_list, self.logits_buffer, self.vocab_size)
+        logits = run_prefill_with_logits(self.model_ptr, input_list,
+                                         self.logits_buffer, self.vocab_size)
     else:
-        logits = run_decode_with_logits(self.model_ptr, input_list[0], self.logits_buffer, self.vocab_size)
+        logits = run_decode_with_logits(self.model_ptr, input_list[0],
+                                        self.logits_buffer, self.vocab_size)
 
     return CausalLMOutputWithPast(
         loss=None,

diff --git a/python/llm/src/ipex_llm/transformers/npu_models/npu_llm_cpp.py b/python/llm/src/ipex_llm/transformers/npu_models/npu_llm_cpp.py
@@ -60,10 +60,12 @@ def get_shared_lib_info(lib_base_name: str):
 _lib.reset.argtypes = [ctypes.c_void_p]
 _lib.reset.restype = None
 
-_lib.run_prefill_with_logits.argtypes = [ctypes.c_void_p, ctypes.POINTER(ctypes.c_int), ctypes.c_int, ctypes.POINTER(ctypes.c_float), ctypes.c_int]
+_lib.run_prefill_with_logits.argtypes = [ctypes.c_void_p, ctypes.POINTER(ctypes.c_int), ctypes.c_int,
+                                         ctypes.POINTER(ctypes.c_float), ctypes.c_int]
 _lib.run_prefill_with_logits.restype = None
 
-_lib.run_decode_with_logits.argtypes = [ctypes.c_void_p, ctypes.c_int, ctypes.POINTER(ctypes.c_float), ctypes.c_int]
+_lib.run_decode_with_logits.argtypes = [ctypes.c_void_p, ctypes.c_int,
+                                        ctypes.POINTER(ctypes.c_float), ctypes.c_int]
 _lib.run_decode_with_logits.restype = None