Fix mem fraction static for nightly tests (#11076)
This commit is contained in:
@@ -19,7 +19,6 @@ from sglang.srt.utils import get_bool_env_var
|
||||
|
||||
_SAMPLING_EPS = 1e-6
|
||||
TOP_K_ALL = 1 << 30
|
||||
DEFAULT_SAMPLING_SEED = 42
|
||||
|
||||
|
||||
class SamplingParams:
|
||||
@@ -56,7 +55,7 @@ class SamplingParams:
|
||||
custom_params: Optional[Dict[str, Any]] = None,
|
||||
stream_interval: Optional[int] = None,
|
||||
logit_bias: Optional[Dict[str, float]] = None,
|
||||
sampling_seed: Optional[int] = None,
|
||||
sampling_seed: int = 42,
|
||||
) -> None:
|
||||
self.max_new_tokens = max_new_tokens
|
||||
self.stop_strs = stop
|
||||
@@ -84,13 +83,6 @@ class SamplingParams:
|
||||
self.custom_params = custom_params
|
||||
self.stream_interval = stream_interval
|
||||
self.logit_bias = logit_bias
|
||||
# Used for deterministic sampling
|
||||
if (
|
||||
get_bool_env_var("SGLANG_ENABLE_DETERMINISTIC_INFERENCE")
|
||||
and sampling_seed is None
|
||||
):
|
||||
# If deterministic inference is enabled and sampling_seed is not set, use the default seed
|
||||
sampling_seed = DEFAULT_SAMPLING_SEED
|
||||
self.sampling_seed = sampling_seed
|
||||
|
||||
# Process some special cases
|
||||
|
||||
Reference in New Issue
Block a user