From 31d05f7edba57751976e329f3d4318383ae5593a Mon Sep 17 00:00:00 2001 From: Woosuk Kwon Date: Mon, 1 Apr 2024 07:07:57 +0000 Subject: [PATCH] yapf --- vllm/executor/tpu_executor.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/vllm/executor/tpu_executor.py b/vllm/executor/tpu_executor.py index 76d13f4017477..80e820209ccf7 100644 --- a/vllm/executor/tpu_executor.py +++ b/vllm/executor/tpu_executor.py @@ -65,10 +65,10 @@ class TPUExecutor(ExecutorBase): """ # Get the maximum number of blocks that can be allocated on TPU. num_tpu_blocks = self.driver_worker.profile_num_available_blocks( - block_size=self.cache_config.block_size, - gpu_memory_utilization=self.cache_config.gpu_memory_utilization, - cache_dtype=self.cache_config.cache_dtype, - ) + block_size=self.cache_config.block_size, + gpu_memory_utilization=self.cache_config.gpu_memory_utilization, + cache_dtype=self.cache_config.cache_dtype, + ) logger.info(f"# TPU blocks: {num_tpu_blocks}") check_block_size_valid(num_tpu_blocks, self.cache_config.block_size,