Skip to content

Commit e9c681c

Browse files
authored
Fix vllm nccl sync error (#132)
1 parent 15f8a8b commit e9c681c

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

trinity/trainer/verl/fsdp_workers.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -27,8 +27,7 @@
2727
import torch
2828
import torch.distributed
2929
import torch.distributed as dist
30-
31-
# import vllm # noqa: F401 ; import vllm to set NCCL_CUMEM_ENABLE automatically.
30+
import vllm # noqa: F401 ; import vllm to set NCCL_CUMEM_ENABLE automatically.
3231
from codetiming import Timer
3332
from omegaconf import DictConfig, OmegaConf, open_dict
3433
from peft import LoraConfig, TaskType, get_peft_model

0 commit comments

Comments
 (0)