Unverified Commit 4a762041 authored by Liangsheng Yin's avatar Liangsheng Yin Committed by GitHub
Browse files

move `environ` into `sglang.srt` to avoid break SRT auto sync. (#10791)

parent ea338676
...@@ -10,7 +10,7 @@ from sglang.srt.utils import kill_process_tree ...@@ -10,7 +10,7 @@ from sglang.srt.utils import kill_process_tree
MOVE_ENVS_WARN = """ MOVE_ENVS_WARN = """
######################################################################## ########################################################################
# For contributors and developers: # # For contributors and developers: #
# Please move environment variable definitions to 'sglang/environ.py' # # Please move environment variable definitions to sglang.srt.environ #
# using the following pattern: # # using the following pattern: #
# SGLANG_XXX = EnvBool(False) # # SGLANG_XXX = EnvBool(False) #
# # # #
......
...@@ -12,8 +12,8 @@ import torch.nn.functional as F ...@@ -12,8 +12,8 @@ import torch.nn.functional as F
import triton import triton
import triton.language as tl import triton.language as tl
from sglang.environ import envs
from sglang.srt.constrained.base_grammar_backend import BaseGrammarObject from sglang.srt.constrained.base_grammar_backend import BaseGrammarObject
from sglang.srt.environ import envs
from sglang.srt.layers.attention.utils import create_flashinfer_kv_indices_triton from sglang.srt.layers.attention.utils import create_flashinfer_kv_indices_triton
from sglang.srt.layers.logits_processor import LogitsProcessorOutput from sglang.srt.layers.logits_processor import LogitsProcessorOutput
from sglang.srt.layers.sampler import apply_custom_logit_processor from sglang.srt.layers.sampler import apply_custom_logit_processor
......
...@@ -195,7 +195,7 @@ _warned_bool_env_var_keys = set() ...@@ -195,7 +195,7 @@ _warned_bool_env_var_keys = set()
def get_bool_env_var(name: str, default: str = "false") -> bool: def get_bool_env_var(name: str, default: str = "false") -> bool:
# FIXME: move your environment variable to sglang.environ # FIXME: move your environment variable to sglang.srt.environ
value = os.getenv(name, default) value = os.getenv(name, default)
value = value.lower() value = value.lower()
...@@ -213,7 +213,7 @@ def get_bool_env_var(name: str, default: str = "false") -> bool: ...@@ -213,7 +213,7 @@ def get_bool_env_var(name: str, default: str = "false") -> bool:
def get_int_env_var(name: str, default: int = 0) -> int: def get_int_env_var(name: str, default: int = 0) -> int:
# FIXME: move your environment variable to sglang.environ # FIXME: move your environment variable to sglang.srt.environ
value = os.getenv(name) value = os.getenv(name)
if value is None or not value.strip(): if value is None or not value.strip():
return default return default
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment