Skip to content

Commit 060e196

Browse files
committed
Revert "resolve template"
This reverts commit 9769b8f.
1 parent 9769b8f commit 060e196

1 file changed

Lines changed: 5 additions & 9 deletions

File tree

ci/benchmarks/partial-conv/evo2_pretrain.yaml

Lines changed: 5 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -45,17 +45,13 @@ script_args:
4545
- model_type: mamba
4646
config_name: hybrid_mamba_8b
4747
script: |-
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"
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
5552
fi
56-
5753
# All ranks wait until install flag file appears
58-
while [ ! -f "$INSTALL_FLAG" ]; do
54+
while [ ! -f $INSTALL_FLAG ]; do
5955
sleep 1
6056
done
6157
WANDB_API_KEY=$BIONEMO_WANDB_API_KEY ${variant}_${model} \

0 commit comments

Comments
 (0)