Update logprob_start_len handling in scheduler (#16240)
This commit is contained in:
@@ -1532,6 +1532,10 @@ class Scheduler(
|
|||||||
self._add_request_to_queue(req)
|
self._add_request_to_queue(req)
|
||||||
return
|
return
|
||||||
|
|
||||||
|
if not recv_req.return_logprob and recv_req.logprob_start_len != -1:
|
||||||
|
# When return_logprob is False, logprob_start_len should be ignored
|
||||||
|
recv_req.logprob_start_len = -1
|
||||||
|
|
||||||
if recv_req.logprob_start_len == -1:
|
if recv_req.logprob_start_len == -1:
|
||||||
if req.is_prefill_only:
|
if req.is_prefill_only:
|
||||||
# For prefill-only requests with logprob_start_len == -1, set logprob_start_len
|
# For prefill-only requests with logprob_start_len == -1, set logprob_start_len
|
||||||
|
|||||||
Reference in New Issue
Block a user