fix line too long

Signed-off-by: yuwenzho <yuwen.zhou@intel.com>
HabanaAI · Aug 29, 2024 · 45f0cc9 · 45f0cc9
1 parent fd1e617
commit 45f0cc9
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/vllm/hpu/cache_ops.py b/vllm/hpu/cache_ops.py
@@ -30,7 +30,8 @@ def reshape_and_cache(key,
     # lots of padding, or are doing warmup.
     # This loop is a workaround for this issue. Please remove it
     # once key_cache.index_put_(indices, offsets), key) works.
-    num_kv_cache_passes = int(math.ceil(num_slots_requested / num_slots_available))
+    num_kv_cache_passes = int(
+        math.ceil(num_slots_requested / num_slots_available))
     for i in range(num_kv_cache_passes):
         start_idx = i * num_slots_available
         end_idx = (i + 1) * num_slots_available
@@ -57,7 +58,8 @@ def prepare_to_cache(cache, slot_mapping):
     # lots of padding, or are doing warmup.
     # This loop is a workaround for this issue. Please remove it
     # once key_cache.index_put_(indices, offsets), key) works.
-    num_kv_cache_passes = int(math.ceil(num_slots_requested / num_slots_available))
+    num_kv_cache_passes = int(
+        math.ceil(num_slots_requested / num_slots_available))
 
     return num_kv_cache_passes, num_slots_available, indices, offsets