[diffusion] refactor: centralize hardware platform detection and streamline environment variable management (#15842)

2025-12-26 22:16:18 +08:00
parent cf34d0ab32
commit 8dc6f0fc4d
12 changed files with 260 additions and 365 deletions
@@ -341,7 +341,9 @@ def initialize_model_parallel(
    """

    if backend is None:
-        backend = envs.get_torch_distributed_backend()
+        from sglang.multimodal_gen.runtime.platforms import current_platform
+
+        backend = current_platform.get_torch_distributed_backend_str()
    # Get world size and rank. Ensure some consistencies.
    assert torch.distributed.is_initialized()
    world_size: int = torch.distributed.get_world_size()