Revert "[Core] Performance: Use list[np.ndarray] instead of list[list… (#28773)

2025-11-14 20:24:00 -08:00
parent edfe498189
commit ac86bff8cb
12 changed files with 76 additions and 102 deletions
--- a/vllm/v1/core/sched/scheduler.py
+++ b/vllm/v1/core/sched/scheduler.py
@@ -1010,8 +1010,8 @@ class Scheduler(SchedulerInterface):
                continue

            req_index = model_runner_output.req_id_to_index[req_id]
-            generated_token_ids: list[int] = (
-                sampled_token_ids[req_index].tolist() if sampled_token_ids else []
+            generated_token_ids = (
+                sampled_token_ids[req_index] if sampled_token_ids else []
            )

            scheduled_spec_token_ids = (