vllm_omni.diffusion.envs ¶

CUDA_HOME: str | None = None

LOCAL_RANK: int = 0

MASTER_ADDR: str = ''

MASTER_PORT: int | None = None

PACKAGES_CHECKER = PackagesEnvChecker()

environment_variables: dict[str, Callable[[], Any]] = {
    "MASTER_ADDR": lambda: getenv("MASTER_ADDR", ""),
    "MASTER_PORT": lambda: (
        int(getenv("MASTER_PORT", "0"))
        if "MASTER_PORT" in environ
        else None
    ),
    "CUDA_HOME": lambda: get("CUDA_HOME", None),
    "LOCAL_RANK": lambda: int(get("LOCAL_RANK", "0")),
}