From 950f3494923c2757330c105368807048b45b223e Mon Sep 17 00:00:00 2001 From: Alexander Matveev Date: Tue, 28 Jan 2025 20:56:21 +0000 Subject: [PATCH] scheduler is clean Signed-off-by: Alexander Matveev --- vllm/v1/core/scheduler.py | 7 ------- 1 file changed, 7 deletions(-) diff --git a/vllm/v1/core/scheduler.py b/vllm/v1/core/scheduler.py index 20f54753b439e..d14819eb720fb 100644 --- a/vllm/v1/core/scheduler.py +++ b/vllm/v1/core/scheduler.py @@ -213,13 +213,6 @@ class Scheduler: num_new_tokens = self.block_size computed_blocks.pop() - # TODO: Remove - # If chunked prefill is not enabled, then breakout of the loop - # when above budget. - # if (not self.scheduler_config.chunked_prefill_enabled - # and num_new_tokens > token_budget): - # break - num_new_tokens = min(num_new_tokens, token_budget) assert num_new_tokens > 0