fix session request with None tokenizer (#16278)

This commit is contained in:
Aurick Qiao
2026-01-14 10:41:08 -08:00
committed by GitHub
parent aa2b4f7661
commit 5b1215d9da
2 changed files with 5 additions and 3 deletions
+3 -1
View File
@@ -1472,7 +1472,9 @@ class Scheduler(
else: else:
# Create a new request from a previous session # Create a new request from a previous session
session = self.sessions[recv_req.session_params.id] session = self.sessions[recv_req.session_params.id]
req = session.create_req(recv_req, self.tokenizer) req = session.create_req(
recv_req, self.tokenizer, self.model_config.vocab_size
)
if isinstance(req.finished_reason, FINISH_ABORT): if isinstance(req.finished_reason, FINISH_ABORT):
self.init_req_max_new_tokens(req) self.init_req_max_new_tokens(req)
self._add_request_to_queue(req) self._add_request_to_queue(req)
@@ -65,7 +65,7 @@ class Session:
self.capacity_of_str_len = capacity_of_str_len self.capacity_of_str_len = capacity_of_str_len
self.req_nodes: Dict[str, SessionReqNode] = {} self.req_nodes: Dict[str, SessionReqNode] = {}
def create_req(self, req: TokenizedGenerateReqInput, tokenizer): def create_req(self, req: TokenizedGenerateReqInput, tokenizer, vocab_size: int):
assert req.session_params is not None assert req.session_params is not None
session_params = req.session_params session_params = req.session_params
@@ -144,7 +144,7 @@ class Session:
return_logprob=req.return_logprob, return_logprob=req.return_logprob,
top_logprobs_num=req.top_logprobs_num, top_logprobs_num=req.top_logprobs_num,
token_ids_logprob=req.token_ids_logprob, token_ids_logprob=req.token_ids_logprob,
vocab_size=tokenizer.vocab_size, vocab_size=vocab_size,
) )
if last_req is not None: if last_req is not None:
new_req.multimodal_inputs = last_req.multimodal_inputs new_req.multimodal_inputs = last_req.multimodal_inputs