Skip to content

Commit 3dc5494

Browse files
committed
Set SCRIPT_DIR to /workspace/code, remove redundant mount
1 parent 5b23fff commit 3dc5494

File tree

1 file changed

+1
-4
lines changed

1 file changed

+1
-4
lines changed

llm_moe_pretraining/nemo/run_deepseek.py

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -55,7 +55,7 @@
5555
logging.basicConfig(level=logging.INFO)
5656
logger = logging.getLogger(__name__)
5757

58-
SCRIPT_DIR = Path(__file__).parent.resolve()
58+
SCRIPT_DIR = Path("/workspace/code")
5959
PRETRAIN_SCRIPT = "pretrain_deepseek_v3_671b.py"
6060

6161
# Inline bash template for Slurm
@@ -142,9 +142,6 @@ def slurm_executor(
142142
env_vars.update(custom_env_vars)
143143
mounts.extend(custom_mounts)
144144

145-
# Mount the script directory
146-
mounts.append(f"{SCRIPT_DIR}:{SCRIPT_DIR}")
147-
148145
# Compute segment for GB200/GB300 if not explicitly provided
149146
if segment is None and num_gpus_per_node == 4:
150147
if nodes <= 18:

0 commit comments

Comments
 (0)