fix format

Signed-off-by: yuwenzho <yuwen.zhou@intel.com>
HabanaAI · Sep 3, 2024 · 2a72462 · 2a72462
1 parent fc48f7e
commit 2a72462
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 1 deletion.
diff --git a/vllm/hpu/cache_ops.py b/vllm/hpu/cache_ops.py
@@ -5,9 +5,11 @@
 # LICENSE file in the root directory of this source tree.
 ###############################################################################
 
+import math
+
 import habana_frameworks.torch as htorch
 import torch
-import math
+
 
 def reshape_and_cache(key,
  value,

diff --git a/vllm/model_executor/models/gpt_bigcode.py b/vllm/model_executor/models/gpt_bigcode.py
@@ -46,6 +46,7 @@
 
 is_hpu = current_platform.is_hpu()
 
+
 class GPTBigCodeAttention(nn.Module):
 
  def __init__(

diff --git a/vllm/model_executor/models/llama.py b/vllm/model_executor/models/llama.py
@@ -57,6 +57,7 @@
 
 is_hpu = current_platform.is_hpu()
 
+
 class LlamaMLP(nn.Module):
 
  def __init__(