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 15f8a8b commit e9c681cCopy full SHA for e9c681c
trinity/trainer/verl/fsdp_workers.py
@@ -27,8 +27,7 @@
27
import torch
28
import torch.distributed
29
import torch.distributed as dist
30
-
31
-# import vllm # noqa: F401 ; import vllm to set NCCL_CUMEM_ENABLE automatically.
+import vllm # noqa: F401 ; import vllm to set NCCL_CUMEM_ENABLE automatically.
32
from codetiming import Timer
33
from omegaconf import DictConfig, OmegaConf, open_dict
34
from peft import LoraConfig, TaskType, get_peft_model
0 commit comments