Merge branch 'main' into increase-slurm-time-limits

kajalj22 · web-flow · commit fe2a94566c83 · 2026-06-10T20:39:44.000-05:00
diff --git a/nemo_rl/models/generation/vllm/vllm_worker.py b/nemo_rl/models/generation/vllm/vllm_worker.py
@@ -106,7 +106,10 @@ def configure_worker(
             init_kwargs["seed"] = seed
             # Need to give each DP group its own vllm cache to address:
             # https://github.com/vllm-project/vllm/issues/18851
-            env_vars["VLLM_CACHE_ROOT"] = os.path.expanduser(f"~/.cache/vllm_{seed}")
+            env_vars["VLLM_CACHE_ROOT"] = (
+                os.environ.get("VLLM_CACHE_ROOT", os.path.expanduser("~/.cache/vllm"))
+                + f"_{seed}"
+            )
 
             # Give each vLLM engine a deterministic starting port for TP/DP
             # rendezvous.  vLLM's _get_open_port() reads VLLM_PORT and
