File tree Expand file tree Collapse file tree
ci/benchmarks/partial-conv Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -45,13 +45,17 @@ script_args:
4545 - model_type : mamba
4646 config_name : hybrid_mamba_8b
4747script : |-
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} \
You can’t perform that action at this time.
0 commit comments