[minor] Make DEFAULT_FORCE_STREAM_INTERVAL configurable via SGLANG_FORCE_STREAM_INTERVAL (#23215)
This commit is contained in:
@@ -266,6 +266,12 @@ class Envs:
|
|||||||
SGLANG_REQ_RUNNING_TIMEOUT = EnvFloat(-1) # in seconds
|
SGLANG_REQ_RUNNING_TIMEOUT = EnvFloat(-1) # in seconds
|
||||||
SGLANG_DISAGGREGATION_BOOTSTRAP_ENTRY_CLEANUP_INTERVAL = EnvInt(120)
|
SGLANG_DISAGGREGATION_BOOTSTRAP_ENTRY_CLEANUP_INTERVAL = EnvInt(120)
|
||||||
SGLANG_SWA_EVICTION_INTERVAL_MULTIPLIER = EnvFloat(1.0)
|
SGLANG_SWA_EVICTION_INTERVAL_MULTIPLIER = EnvFloat(1.0)
|
||||||
|
# For non-streaming requests, the scheduler still flushes intermediate
|
||||||
|
# output batches to the tokenizer manager every N decoded tokens so that
|
||||||
|
# `first_token_time`/TTFT can be recorded. Lower this (e.g. to 1) to get
|
||||||
|
# an accurate TTFT for benchmarking; the upstream default of 50 trades
|
||||||
|
# off some TTFT-metric accuracy for less IPC overhead.
|
||||||
|
SGLANG_FORCE_STREAM_INTERVAL = EnvInt(50)
|
||||||
|
|
||||||
# Test: pd-disaggregation
|
# Test: pd-disaggregation
|
||||||
SGLANG_TEST_PD_DISAGG_BACKEND = EnvStr("mooncake")
|
SGLANG_TEST_PD_DISAGG_BACKEND = EnvStr("mooncake")
|
||||||
|
|||||||
@@ -33,7 +33,9 @@ if TYPE_CHECKING:
|
|||||||
|
|
||||||
logger = logging.getLogger(__name__)
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
DEFAULT_FORCE_STREAM_INTERVAL = 50
|
# How often (in decoded tokens) the scheduler force-flushes an intermediate
|
||||||
|
# output batch for non-streaming requests.
|
||||||
|
DEFAULT_FORCE_STREAM_INTERVAL = envs.SGLANG_FORCE_STREAM_INTERVAL.get()
|
||||||
|
|
||||||
|
|
||||||
class SchedulerOutputProcessorMixin:
|
class SchedulerOutputProcessorMixin:
|
||||||
|
|||||||
Reference in New Issue
Block a user