Support prefill max requests limitation (#14993)
This commit is contained in:
@@ -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",
|
||||
|
||||
Reference in New Issue
Block a user