Skip to content

Commit 7bf1a1f

Browse files
committed
remove image dependence, update image, and add new bionemo checkout script
Signed-off-by: jwilber <jwilber@nvidia.com>
1 parent 29f8cbd commit 7bf1a1f

1 file changed

Lines changed: 1 addition & 29 deletions

File tree

ci/lepton/model_convergence/configs/recipes/codonfm_ptl_te.yaml

Lines changed: 1 addition & 29 deletions
Original file line numberDiff line numberDiff line change
@@ -15,15 +15,7 @@ gpu_type: h100-sxm
1515
resource_shape: "${device_type}.${num_devices}x${gpu_type}"
1616

1717
job_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
8476
attn_input_format: thd
8577
collate_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

Comments
 (0)