Skip to content

Commit 29db947

Browse files
authored
[Fix] all RayEngineWorker actors created at node 0 in RL training (#4107)
1 parent 9305204 commit 29db947

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

lmdeploy/pytorch/engine/mp_engine/ray_engine.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -111,7 +111,7 @@ def _init_ray(self, engine_config: PytorchEngineConfig = None):
111111

112112
def _create_worker(self, model_path: str, engine_config: PytorchEngineConfig = None, **kwargs):
113113
"""Create a Ray worker."""
114-
bundle_id = 0
114+
bundle_id = 0 if len(_envs.ray_external_pg_bundles) == 0 else _envs.ray_external_pg_bundles[0]
115115
scheduling_strategy = PlacementGroupSchedulingStrategy(
116116
placement_group=self.placement_group,
117117
placement_group_capture_child_tasks=True,

0 commit comments

Comments
 (0)