diff --git a/python/sglang/srt/managers/scheduler.py b/python/sglang/srt/managers/scheduler.py index 80e5e4c0f..4ff5845f5 100644 --- a/python/sglang/srt/managers/scheduler.py +++ b/python/sglang/srt/managers/scheduler.py @@ -1472,7 +1472,9 @@ class Scheduler( else: # Create a new request from a previous session 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): self.init_req_max_new_tokens(req) self._add_request_to_queue(req) diff --git a/python/sglang/srt/managers/session_controller.py b/python/sglang/srt/managers/session_controller.py index 4990a5cac..7e545ff87 100644 --- a/python/sglang/srt/managers/session_controller.py +++ b/python/sglang/srt/managers/session_controller.py @@ -65,7 +65,7 @@ class Session: self.capacity_of_str_len = capacity_of_str_len 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 session_params = req.session_params @@ -144,7 +144,7 @@ class Session: return_logprob=req.return_logprob, top_logprobs_num=req.top_logprobs_num, token_ids_logprob=req.token_ids_logprob, - vocab_size=tokenizer.vocab_size, + vocab_size=vocab_size, ) if last_req is not None: new_req.multimodal_inputs = last_req.multimodal_inputs