We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 22af21c commit 4a2e126Copy full SHA for 4a2e126
nemo_reinforcer/models/generation/vllm.py
@@ -109,6 +109,8 @@ def configure_worker(
109
init_kwargs["fraction_of_gpus"] = num_gpus
110
111
env_vars["VLLM_ENABLE_V1_MULTIPROCESSING"] = "0"
112
+ # Skip vllm P2P check and rely on driver to report peer to peer capability.
113
+ env_vars["VLLM_SKIP_P2P_CHECK"] = "1"
114
115
return resources, env_vars, init_kwargs
116
0 commit comments