We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 24eac4b commit 8893c78Copy full SHA for 8893c78
tests/distributed/eplb_utils.py
@@ -1,11 +1,11 @@
1
# SPDX-License-Identifier: Apache-2.0
2
# SPDX-FileCopyrightText: Copyright contributors to the vLLM project
3
4
-import torch.multiprocessing as mp
5
import os
6
import random
7
8
import torch
+import torch.multiprocessing as mp
9
10
from vllm.distributed.parallel_state import (
11
init_distributed_environment,
@@ -46,4 +46,4 @@ def set_env_vars_and_device(env: dict[str, str]) -> None:
46
47
# Ensure each worker process has the same random seed
48
random.seed(42)
49
- torch.manual_seed(42)
+ torch.manual_seed(42)
0 commit comments