fix session request with None tokenizer (#16278)
This commit is contained in:
@@ -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
|
||||||
|
|||||||
Reference in New Issue
Block a user