[diffusion] fix: remove multimodal_gen redundant get_bool_env_var func (#13583)
Co-authored-by: Mick <mickjagger19@icloud.com>
This commit is contained in:
@@ -308,30 +308,6 @@ def set_cuda_arch():
|
|||||||
os.environ["TORCH_CUDA_ARCH_LIST"] = f"{arch}{'+PTX' if arch == '9.0' else ''}"
|
os.environ["TORCH_CUDA_ARCH_LIST"] = f"{arch}{'+PTX' if arch == '9.0' else ''}"
|
||||||
|
|
||||||
|
|
||||||
def get_bool_env_var(env_var_name: str, default: str | bool = "false") -> bool:
|
|
||||||
raw_value = os.getenv(env_var_name, None)
|
|
||||||
if raw_value is None:
|
|
||||||
raw_value = str(default)
|
|
||||||
|
|
||||||
value_str = str(raw_value).strip().lower()
|
|
||||||
truthy = {"1", "true", "yes", "y", "t", "on"}
|
|
||||||
falsy = {"0", "false", "no", "n", "f", "off", ""}
|
|
||||||
|
|
||||||
if value_str in truthy:
|
|
||||||
return True
|
|
||||||
if value_str in falsy:
|
|
||||||
return False
|
|
||||||
|
|
||||||
default_bool = str(default).strip().lower() in truthy
|
|
||||||
logger.warning(
|
|
||||||
"Unrecognized boolean for %s=%r; falling back to default=%r",
|
|
||||||
env_var_name,
|
|
||||||
raw_value,
|
|
||||||
default_bool,
|
|
||||||
)
|
|
||||||
return default_bool
|
|
||||||
|
|
||||||
|
|
||||||
def is_flashinfer_available():
|
def is_flashinfer_available():
|
||||||
"""
|
"""
|
||||||
Check whether flashinfer is available.
|
Check whether flashinfer is available.
|
||||||
@@ -348,12 +324,11 @@ _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.srt.environ
|
|
||||||
value = os.getenv(name, default)
|
value = os.getenv(name, default)
|
||||||
value = value.lower()
|
value = str(value).strip().lower()
|
||||||
|
|
||||||
truthy_values = ("true", "1")
|
truthy_values = {"1", "true", "yes", "y", "t", "on"}
|
||||||
falsy_values = ("false", "0")
|
falsy_values = {"0", "false", "no", "n", "f", "off", ""}
|
||||||
|
|
||||||
if (value not in truthy_values) and (value not in falsy_values):
|
if (value not in truthy_values) and (value not in falsy_values):
|
||||||
if value not in _warned_bool_env_var_keys:
|
if value not in _warned_bool_env_var_keys:
|
||||||
|
|||||||
Reference in New Issue
Block a user