Skip to content

Jwilber/cd image up4 #1569

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
jwilber wants to merge 2 commits into
base: main
Choose a base branch
from
Open

Jwilber/cd image up4 #1569

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
29f8cbd
bump image
jwilber May 4, 2026
7bf1a1f
remove image dependence, update image, and add new bionemo checkout s…
jwilber May 4, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
30 changes: 1 addition & 29 deletions ci/lepton/model_convergence/configs/recipes/codonfm_ptl_te.yaml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -15,15 +15,7 @@ gpu_type: h100-sxm
resource_shape: "${device_type}.${num_devices}x${gpu_type}"

job_name: "${sanitize:codonfm_ptl_test}8gpu"
branch: jwilber/codon-fm-ci

############################################################
# Container Runtime
# Defines the base Docker image and registry auth needed
############################################################
container:
image: nvcr.io/nvidian/cvai_bnmo_trng/codon_fm_te_bnmo_10292025_multinode
registry_auth: lepton-nvidia-nvcrio-jonathan
branch: main

############################################################
# kratos info: where to log data
Expand Down Expand Up @@ -84,26 +76,6 @@ split_name_prefix: nopathogen
attn_input_format: thd
collate_fn: thd

############################################################
# Checkout Script
# Standardized script to clone the BioNeMo repository and install
# dependencies before the training run starts. Child configs can
# inherit and reuse this logic without modification.
############################################################
checkout_script: |
cd /workspace/codonfm
git clone https://github.com/NVIDIA/bionemo-framework.git
cd bionemo-framework
if [ -n "${commit_sha}" ]; then
echo "Checking out commit: ${commit_sha}"
git checkout "${commit_sha}"
elif [ "${branch}" != "main" ]; then
echo "Checking out branch: ${branch}"
git checkout "${branch}"
fi
cd bionemo-recipes/recipes/${recipe_subdir}
pip install -e .

############################################################
# run script
# This gets called right after `checkout_script` in the base config.
Expand Down
2 changes: 1 addition & 1 deletion ci/lepton/scdl_performance/configs/scdl.yaml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -12,7 +12,7 @@ template_type: scdl_performance
# Defines the base Docker image and registry auth needed
############################################################
container:
image: nvcr.io/nvidia/pytorch:26.02-py3
image: nvcr.io/nvidia/pytorch:26.04-py3
registry_auth: lepton-nvidia-jwilber

############################################################
Expand Down
Loading