Skip to content

Commit 9769b8f

Browse files
committed
resolve template
Signed-off-by: Jared Wilber <jwilber@nvidia.com>
1 parent 30c32df commit 9769b8f

1 file changed

Lines changed: 9 additions & 5 deletions

File tree

ci/benchmarks/partial-conv/evo2_pretrain.yaml

Lines changed: 9 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -45,13 +45,17 @@ script_args:
4545
- model_type: mamba
4646
config_name: hybrid_mamba_8b
4747
script: |-
48-
INSTALL_FLAG="/tmp/install_done_${{SLURMD_NODENAME}}";
49-
if [ "$SLURM_LOCALID" = "0" ]; then
50-
pip install ${file_name_wheel} --index-url ${artefacts_url}
51-
touch $INSTALL_FLAG
48+
INSTALL_FLAG="/tmp/install_done_{{ '${SLURMD_NODENAME}' }}"
49+
50+
# Same idea for SLURM_LOCALID — emit a literal $SLURM_LOCALID:
51+
if [ "{{ '$SLURM_LOCALID' }}" = "0" ]; then
52+
# artefacts_url already has the token resolved by the templater
53+
pip install "${file_name_wheel}" --index-url "${artefacts_url}"
54+
touch "$INSTALL_FLAG"
5255
fi
56+
5357
# All ranks wait until install flag file appears
54-
while [ ! -f $INSTALL_FLAG ]; do
58+
while [ ! -f "$INSTALL_FLAG" ]; do
5559
sleep 1
5660
done
5761
WANDB_API_KEY=$BIONEMO_WANDB_API_KEY ${variant}_${model} \

0 commit comments

Comments
 (0)