@@ -15,15 +15,7 @@ gpu_type: h100-sxm
1515resource_shape : " ${device_type}.${num_devices}x${gpu_type}"
1616
1717job_name : " ${sanitize:codonfm_ptl_test}8gpu"
18- branch : jwilber/codon-fm-ci
19-
20- # ###########################################################
21- # Container Runtime
22- # Defines the base Docker image and registry auth needed
23- # ###########################################################
24- container :
25- image : nvcr.io/nvidian/cvai_bnmo_trng/codon_fm_te_bnmo_10292025_multinode
26- registry_auth : lepton-nvidia-nvcrio-jonathan
18+ branch : main
2719
2820# ###########################################################
2921# kratos info: where to log data
@@ -84,26 +76,6 @@ split_name_prefix: nopathogen
8476attn_input_format : thd
8577collate_fn : thd
8678
87- # ###########################################################
88- # Checkout Script
89- # Standardized script to clone the BioNeMo repository and install
90- # dependencies before the training run starts. Child configs can
91- # inherit and reuse this logic without modification.
92- # ###########################################################
93- checkout_script : |
94- cd /workspace/codonfm
95- git clone https://github.com/NVIDIA/bionemo-framework.git
96- cd bionemo-framework
97- if [ -n "${commit_sha}" ]; then
98- echo "Checking out commit: ${commit_sha}"
99- git checkout "${commit_sha}"
100- elif [ "${branch}" != "main" ]; then
101- echo "Checking out branch: ${branch}"
102- git checkout "${branch}"
103- fi
104- cd bionemo-recipes/recipes/${recipe_subdir}
105- pip install -e .
106-
10779# ###########################################################
10880# run script
10981# This gets called right after `checkout_script` in the base config.
0 commit comments