diff --git a/config.json b/config.json
index d119fe3..b45e3db 100644
--- a/config.json
+++ b/config.json
@@ -4,7 +4,7 @@
   ],
   "attention_probs_dropout_prob": 0.0,
   "classifier_dropout_prob": 0.1,
-  "decoder_hidden_size": 768,
+  "decoder_hidden_size": 64,
   "depths": [
     3,
     8,
@@ -18,6 +18,7 @@
     16
   ],
   "drop_path_rate": 0.1,
+  "head_in_index": -1,
   "hidden_act": "gelu",
   "hidden_dropout_prob": 0.0,
   "hidden_sizes": [
@@ -27,7 +28,6 @@
     512
   ],
   "image_size": 224,
-  "in_index": -1,
   "initializer_range": 0.02,
   "layer_norm_eps": 1e-06,
   "max_depth": 10,
diff --git a/pytorch_model.bin b/pytorch_model.bin
index 9964a35..ddd9cec 100644
--- a/pytorch_model.bin
+++ b/pytorch_model.bin
@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:f8c84ea77e1b4019653704a644944bcf88fba27588e43cf1582c511b2327d821
-size 245256489
+oid sha256:c353b12e5f4e5fafd0053d1c19bf60dfb1562b74db4728c6920a2cf44782b603
+size 245258793