Skip to content

Commit

Permalink
🎨 fmt
Browse files Browse the repository at this point in the history
Signed-off-by: Prashant Gupta <[email protected]>
  • Loading branch information
prashantgupta24 committed Nov 22, 2024
1 parent cb8fc93 commit 8a8a07f
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions vllm/core/scheduler.py
Original file line number Diff line number Diff line change
Expand Up @@ -1070,9 +1070,9 @@ def _schedule_prefills(
continue

num_new_seqs = seq_group.get_max_num_running_seqs()
# quick budget check
if num_new_tokens == 0 or not budget.can_schedule(
num_new_tokens=num_new_tokens, num_new_seqs=num_new_seqs):
if (num_new_tokens == 0
or not budget.can_schedule(num_new_tokens=num_new_tokens,
num_new_seqs=num_new_seqs)):
break

# Can schedule this request.
Expand Down

0 comments on commit 8a8a07f

Please sign in to comment.