From 29f8cbd190b33483584496977a2731766c8c606f Mon Sep 17 00:00:00 2001
From: jwilber <jwilber@nvidia.com>
Date: Mon, 4 May 2026 12:29:31 -0700
Subject: [PATCH 1/2] bump image

Signed-off-by: jwilber <jwilber@nvidia.com>
---
 ci/lepton/scdl_performance/configs/scdl.yaml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/ci/lepton/scdl_performance/configs/scdl.yaml b/ci/lepton/scdl_performance/configs/scdl.yaml
index 16325ab7db..603182faf9 100644
--- a/ci/lepton/scdl_performance/configs/scdl.yaml
+++ b/ci/lepton/scdl_performance/configs/scdl.yaml
@@ -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
 
 ############################################################

From 7bf1a1fb82dd328ddde7b67a05f6a8d3092a5800 Mon Sep 17 00:00:00 2001
From: jwilber <jwilber@nvidia.com>
Date: Mon, 4 May 2026 12:46:21 -0700
Subject: [PATCH 2/2] remove image dependence, update image, and add new
 bionemo checkout script

Signed-off-by: jwilber <jwilber@nvidia.com>
---
 .../configs/recipes/codonfm_ptl_te.yaml       | 30 +------------------
 1 file changed, 1 insertion(+), 29 deletions(-)

diff --git a/ci/lepton/model_convergence/configs/recipes/codonfm_ptl_te.yaml b/ci/lepton/model_convergence/configs/recipes/codonfm_ptl_te.yaml
index 7d75eb9f92..c6b28110eb 100644
--- a/ci/lepton/model_convergence/configs/recipes/codonfm_ptl_te.yaml
+++ b/ci/lepton/model_convergence/configs/recipes/codonfm_ptl_te.yaml
@@ -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
@@ -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.