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.
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
 
 ############################################################