[Bugfix][Core]remove preempted_req in req_to_new_blocks and num_scheduled_tokens

Signed-off-by: quanliu <18646313696@163.com>

Author: quanliu1991

vllm/v1/core/sched/scheduler.py

@@ -273,6 +273,9 @@ def schedule(self) -> SchedulerOutput:
                     self.running.remove(preempted_req)
                     if preempted_req in scheduled_running_reqs:
                         scheduled_running_reqs.remove(preempted_req)
+                        token_budget += num_scheduled_tokens[preempted_req.request_id]
+                        req_to_new_blocks.pop(preempted_req.request_id)
+                        num_scheduled_tokens.pop(preempted_req.request_id)
                 else:
                     preempted_req = self.running.pop()