diff --git a/configs/train_2_1/vitl16/cooldown-256px-64f.yaml b/configs/train_2_1/vitl16/cooldown-256px-64f.yaml
index 73308856..6a74c2ea 100644
--- a/configs/train_2_1/vitl16/cooldown-256px-64f.yaml
+++ b/configs/train_2_1/vitl16/cooldown-256px-64f.yaml
@@ -118,7 +118,7 @@ model:
   model_name: vit_large
   n_registers: 0
   normalize_predictor: false
-  pred_depth: 24
+  pred_depth: 12
   pred_embed_dim: 384
   pred_num_heads: 12
   uniform_power: true
diff --git a/configs/train_2_1/vitl16/pretrain-256px-16f.yaml b/configs/train_2_1/vitl16/pretrain-256px-16f.yaml
index 3988cf9a..6ffe061e 100644
--- a/configs/train_2_1/vitl16/pretrain-256px-16f.yaml
+++ b/configs/train_2_1/vitl16/pretrain-256px-16f.yaml
@@ -120,7 +120,7 @@ model:
   n_registers: 0
   n_registers_predictor: 0
   normalize_predictor: false
-  pred_depth: 24
+  pred_depth: 12
   pred_embed_dim: 384
   pred_is_causal: false
   pred_local_window: