Support prefill max requests limitation (#14993)

This commit is contained in:
fzyzcjy
2025-12-14 10:52:10 +08:00
committed by GitHub
parent 69cfb17b9a
commit 168a31eb00
3 changed files with 14 additions and 0 deletions
+7
View File
@@ -290,6 +290,7 @@ class ServerArgs:
chunked_prefill_size: Optional[int] = None
enable_dynamic_chunking: bool = False
max_prefill_tokens: int = 16384
prefill_max_requests: Optional[int] = None
schedule_policy: str = "fcfs"
enable_priority_scheduling: bool = False
abort_on_priority_when_disabled: bool = False
@@ -2516,6 +2517,12 @@ class ServerArgs:
default=ServerArgs.chunked_prefill_size,
help="The maximum number of tokens in a chunk for the chunked prefill. Setting this to -1 means disabling chunked prefill.",
)
parser.add_argument(
"--prefill-max-requests",
type=int,
default=ServerArgs.prefill_max_requests,
help="The maximum number of requests in a prefill batch. If not specified, there is no limit.",
)
parser.add_argument(
"--enable-dynamic-chunking",
action="store_true",