Skip to content

Commit 2b23e50

Browse files
committed
undo the colocation
1 parent f7ed526 commit 2b23e50

File tree

1 file changed

+8
-3
lines changed

1 file changed

+8
-3
lines changed

src/forge/actors/policy.py

Lines changed: 8 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -155,9 +155,14 @@ async def launch( # pyright: ignore[reportIncompatibleMethodOverride]
155155
policy_proc_config = copy(process_config)
156156
policy_proc_config.procs = 1
157157
policy_proc_config.with_gpus = False
158-
policy_proc = await get_proc_mesh(
159-
process_config=policy_proc_config, host_mesh=host_mesh
160-
)
158+
159+
# TODO - not working yet, delete this once debugged
160+
policy_proc_config.hosts = None
161+
policy_proc = await get_proc_mesh(process_config=policy_proc_config)
162+
163+
# policy_proc = await get_proc_mesh(
164+
# process_config=policy_proc_config, host_mesh=host_mesh
165+
# )
161166

162167
if isinstance(engine_args, Mapping):
163168
engine_args = EngineArgs(**engine_args)

0 commit comments

Comments
 (0)