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,17 +45,13 @@ script_args:
4545 - model_type : mamba
4646 config_name : hybrid_mamba_8b
4747script : |-
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} \
You can’t perform that action at this time.
0 commit comments