[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:
1 parent
12e9280010
commit
b656fcf45b
8 files changed
+28
-42
No files matched your search
@@ -7,14 +7,14 @@ import gc
|
||||
|
||||
import pytest
|
||||
import torch.distributed as dist
|
||||
from vllm.utils.network_utils import get_file_store_init_method
|
||||
|
||||
from tests.helpers.runtime import get_distributed_init_method
|
||||
from vllm_omni.platforms import current_omni_platform
|
||||
|
||||
|
||||
@pytest.fixture(scope="module")
|
||||
def dist_group():
|
||||
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:
|
||||
|
||||
Reference in new issue
Block a user