From 83393b3641b887c7e31fe40e05ef9c126c342eb8 Mon Sep 17 00:00:00 2001 From: Danielle Pintz Date: Wed, 3 Dec 2025 01:37:39 -0800 Subject: [PATCH] upd --- .meta/mast/client_bootstrap.sh | 2 +- src/forge/controller/launcher.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/.meta/mast/client_bootstrap.sh b/.meta/mast/client_bootstrap.sh index e3ccff8fe..9b8a704c3 100755 --- a/.meta/mast/client_bootstrap.sh +++ b/.meta/mast/client_bootstrap.sh @@ -39,7 +39,7 @@ if [ -z "$WORKSPACE_DIR" ] || [ ! -d "$WORKSPACE_DIR" ]; then WORKSPACE_DIR="$CONDA_PREFIX" fi -cd "$WORKSPACE_DIR/forge" +cd "$WORKSPACE_DIR/torchforge" # Execute the client training script with all passed arguments exec python -X faulthandler .meta/mast/main.py "$@" diff --git a/src/forge/controller/launcher.py b/src/forge/controller/launcher.py index 0a6f32c07..569b28b15 100644 --- a/src/forge/controller/launcher.py +++ b/src/forge/controller/launcher.py @@ -206,7 +206,7 @@ def __init__( self.timeout_sec = 1 * 60 * 60 # Kill the job if idle for 1 hour self.user = getpass.getuser() self.work_dir = f"/home/{self.user}" - self.edittable_workspaces = ["forge"] + self.edittable_workspaces = ["torchforge"] self.remote_work_dir = "/packages/monarch_default_workspace/workspace/" self.editable_workspace_paths = [ f"{self.work_dir}/{workspace}" for workspace in self.edittable_workspaces @@ -367,7 +367,7 @@ def _create_client_role(self, appdef: specs.AppDef) -> specs.Role: # Override with client-specific configuration client_role.name = "client" # Use the bootstrap script as entrypoint - client_role.entrypoint = "workspace/forge/.meta/mast/client_bootstrap.sh" + client_role.entrypoint = "workspace/torchforge/.meta/mast/client_bootstrap.sh" # Build args for the client role (passed to the bootstrap script) # These args will be passed to client_bootstrap.sh which forwards them to main.py