allow requests with exactly context_len total tokens (#22546)
Co-authored-by: Ethan (Yusheng) Su <yushengsu.thu@gmail.com>
This commit is contained in:
@@ -842,7 +842,7 @@ class TokenizerManager(TokenizerControlMixin, TokenizerManagerScoreMixin):
|
|||||||
if (
|
if (
|
||||||
self.validate_total_tokens
|
self.validate_total_tokens
|
||||||
and max_new_tokens is not None
|
and max_new_tokens is not None
|
||||||
and (max_new_tokens + input_token_num) >= _max_req_len
|
and (max_new_tokens + input_token_num) > _max_req_len
|
||||||
):
|
):
|
||||||
if self.server_args.allow_auto_truncate:
|
if self.server_args.allow_auto_truncate:
|
||||||
logger.warning(
|
logger.warning(
|
||||||
|
|||||||
Reference in New Issue
Block a user