diff --git a/examples/rlix/run_miles_dual.py b/examples/rlix/run_miles_dual.py index f637de0944a..33778d07648 100644 --- a/examples/rlix/run_miles_dual.py +++ b/examples/rlix/run_miles_dual.py @@ -295,6 +295,11 @@ def _build_pipeline( "TMS_INIT_ENABLE_CPU_BACKUP", "CUDA_DEVICE_MAX_CONNECTIONS", "NCCL_NVLS_ENABLE", + # Resize-chain timeout knobs read inside the MilesCoordinator + # actor process (rlops/rlix#35); without explicit forwarding a + # driver-side export never reaches the actor on multi-node. + "RLIX_RESIZE_LOCK_TIMEOUT_S", + "MILES_RESIZE_RPC_TIMEOUT_S", ): if (_v := os.environ.get(_k)) is not None: pipeline_runtime_env_vars[_k] = _v diff --git a/examples/rlix/run_miles_rlix.py b/examples/rlix/run_miles_rlix.py index 681c11f24fb..e4eac8d210b 100644 --- a/examples/rlix/run_miles_rlix.py +++ b/examples/rlix/run_miles_rlix.py @@ -184,6 +184,11 @@ class MilesPipelineConfig: "TMS_INIT_ENABLE_CPU_BACKUP", "CUDA_DEVICE_MAX_CONNECTIONS", "NCCL_NVLS_ENABLE", + # Resize-chain timeout knobs read inside the MilesCoordinator + # actor process (rlops/rlix#35); without explicit forwarding a + # driver-side export never reaches the actor on multi-node. + "RLIX_RESIZE_LOCK_TIMEOUT_S", + "MILES_RESIZE_RPC_TIMEOUT_S", ): if (_v := os.environ.get(_k)) is not None: pipeline_runtime_env_vars[_k] = _v