Skip to content

Commit 1b215ee

Browse files
yapf
Signed-off-by: baonudesifeizhai <[email protected]>
1 parent 66e4d6d commit 1b215ee

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

vllm/v1/worker/gpu_model_runner.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -888,9 +888,9 @@ def _get_encoder_seq_lens(
888888
# encoder lengths for inputs scheduled in this batch
889889
num_reqs_int = int(num_reqs)
890890
if num_reqs_int == 0:
891-
return np.zeros((0,), dtype=np.int32)
891+
return np.zeros((0, ), dtype=np.int32)
892892

893-
encoder_seq_lens = np.zeros((num_reqs_int,), dtype=np.int32)
893+
encoder_seq_lens = np.zeros((num_reqs_int, ), dtype=np.int32)
894894
for req_id in scheduler_output.scheduled_encoder_inputs:
895895
req_index = self.input_batch.req_id_to_index[req_id]
896896
encoder_seq_lens[req_index] = self.max_encoder_len

0 commit comments

Comments
 (0)