Use more general heuristics to set the default value of --mem-fraction-static (#10975)
Co-authored-by: sglang-bot <sglangbot@gmail.com>
This commit is contained in:
@@ -35,6 +35,7 @@ else:
|
||||
Image = Any
|
||||
|
||||
|
||||
# Parameters for a session
|
||||
@dataclass
|
||||
class SessionParams:
|
||||
id: Optional[str] = None
|
||||
@@ -84,8 +85,6 @@ class GenerateReqInput:
|
||||
sampling_params: Optional[Union[List[Dict], Dict]] = None
|
||||
# The request id.
|
||||
rid: Optional[Union[List[str], str]] = None
|
||||
# Extra key for classifying the request (e.g. cache_salt)
|
||||
extra_key: Optional[Union[List[str], str]] = None
|
||||
# Whether to return logprobs.
|
||||
return_logprob: Optional[Union[List[bool], bool]] = None
|
||||
# If return logprobs, the start location in the prompt for returning logprobs.
|
||||
@@ -134,18 +133,23 @@ class GenerateReqInput:
|
||||
# Conversation id used for tracking requests
|
||||
conversation_id: Optional[str] = None
|
||||
|
||||
# (Deprecated, please use custom_labels) Label for the request
|
||||
label: Optional[str] = None
|
||||
|
||||
# Priority for the request
|
||||
priority: Optional[int] = None
|
||||
|
||||
# Image gen grpc migration
|
||||
return_bytes: bool = False
|
||||
# Extra key for classifying the request (e.g. cache_salt)
|
||||
extra_key: Optional[Union[List[str], str]] = None
|
||||
|
||||
# Whether to disallow logging for this request (e.g. due to ZDR)
|
||||
no_logs: bool = False
|
||||
|
||||
# For custom metric labels
|
||||
custom_labels: Optional[Dict[str, str]] = None
|
||||
|
||||
# (Deprecated, please use custom_labels) Label for the request
|
||||
label: Optional[str] = None
|
||||
# (Internal) Whether to return bytes for image generation
|
||||
return_bytes: bool = False
|
||||
|
||||
def contains_mm_input(self) -> bool:
|
||||
return (
|
||||
has_valid_data(self.image_data)
|
||||
@@ -544,8 +548,11 @@ class GenerateReqInput:
|
||||
self.data_parallel_rank if self.data_parallel_rank is not None else None
|
||||
),
|
||||
conversation_id=self.conversation_id,
|
||||
label=self.label,
|
||||
priority=self.priority,
|
||||
extra_key=self.extra_key,
|
||||
no_logs=self.no_logs,
|
||||
custom_labels=self.custom_labels,
|
||||
label=self.label,
|
||||
return_bytes=self.return_bytes,
|
||||
)
|
||||
|
||||
@@ -602,21 +609,23 @@ class TokenizedGenerateReqInput:
|
||||
# For dp balance
|
||||
dp_balance_id: int = -1
|
||||
|
||||
# Label for the request
|
||||
label: Optional[str] = None
|
||||
|
||||
# Priority for the request
|
||||
priority: Optional[int] = None
|
||||
|
||||
# Extra key for classifying the request (e.g. cache_salt)
|
||||
extra_key: Optional[str] = None
|
||||
|
||||
# Image gen grpc migration
|
||||
return_bytes: bool = False
|
||||
# Whether to disallow logging for this request (e.g. due to ZDR)
|
||||
no_logs: bool = False
|
||||
|
||||
# tracing context
|
||||
trace_context: Optional[Dict] = None
|
||||
|
||||
# (Deprecated, please use custom_labels) Label for the request
|
||||
label: Optional[str] = None
|
||||
# (Internal) Whether to return bytes for image generation
|
||||
return_bytes: bool = False
|
||||
|
||||
|
||||
@dataclass
|
||||
class BatchTokenizedGenerateReqInput:
|
||||
|
||||
Reference in New Issue
Block a user