Update config.json

config.json

@@ -5,6 +5,16 @@
   "initializer_range": 0.02,
   "mask_decoder_config": {
     "model_type": ""
+        "transformer_dim": 256,
+    "iou_head_depth": 3, 
+    "iou_head_hidden_dim": 256, 
+    "num_multimask_outputs": 3, 
+    "transformer": {
+      "depth": 2, 
+      "embedding_dim": 256, 
+      "mlp_dim": 2048, 
+      "num_heads": 8
+    }
   },
   "model_type": "sam",
   "prompt_encoder_config": {
@@ -13,8 +23,22 @@
   "torch_dtype": "float32",
   "transformers_version": "4.37.0.dev0",
   "vision_config": {
-    "hidden_size": 384,
-    "mlp_dim": 1536,
+    "img_size": 1024, 
+    "in_chans": 3, 
+    "num_classes": 1000, 
+    "embed_dims": [64, 128, 160, 320], 
+    "depths": [2, 2, 6, 2], 
+    "num_heads": [2, 4, 5, 10], 
+    "window_sizes": [7, 7, 14, 7], 
+    "mlp_ratio": 4.0, 
+    "drop_rate": 0.0, 
+    "drop_path_rate": 0.0, 
+    "use_checkpoint": false, 
+    "mbconv_expand_ratio": 4.0, 
+    "local_conv_size": 3, 
+    "layer_lr_decay": 0.8
     "model_type": ""
   }
-}
+  "pixel_mean": [123.675, 116.28, 103.53], 
+  "pixel_std": [58.395, 57.12, 57.375]
+}