diff --git a/configs/wilds_vitb16-448_ep600.yaml b/configs/wilds_vitb16-448_ep600.yaml
index e4c88e5..27f06ec 100644
--- a/configs/wilds_vitb16-448_ep600.yaml
+++ b/configs/wilds_vitb16-448_ep600.yaml
@@ -35,7 +35,7 @@ meta:
   load_checkpoint: false
   model_name: vit_base
   embed_dim: 768
-  pred_depth: 12
+  pred_depth: 6
   pred_emb_dim: 384
   read_checkpoint: null
   use_bfloat16: true