[Tests] Use get_file_store_init_method helper from vllm (#7528)

Signed-off-by: Nick Cao <ncao@redhat.com>
Co-authored-by: Codex <noreply@openai.com>
This commit is contained in:
Nick CaoandCodex authored and GitHub committed 2026-09-28 14:22:25 -04:00
1 parent 12e9280010
commit b656fcf45b
8 files changed
+28 -42

No files matched your search

+2 -2
View File
@@ -9,8 +9,8 @@ import torch
import torch.distributed as dist
import torch.nn as nn
from torch.distributed.tensor import DeviceMesh, DTensor
from vllm.utils.network_utils import get_file_store_init_method
from tests.helpers.runtime import get_distributed_init_method
from vllm_omni.diffusion.data import DiffusionParallelConfig
from vllm_omni.diffusion.distributed import hsdp as hsdp_module
from vllm_omni.diffusion.distributed.hsdp import (
@@ -43,7 +43,7 @@ def cpu_process_group():
yield
return
dist.init_process_group("gloo", rank=0, world_size=1, init_method=get_distributed_init_method())
dist.init_process_group("gloo", rank=0, world_size=1, init_method=get_file_store_init_method())
try:
yield
finally: