Upload config

config.json

@@ -1,15 +1,14 @@
 {
-  "contrastive_loss": false,
-  "decoder_depth": 2,
-  "decoder_dim": 1024,
-  "decoder_dim_head": 64,
-  "decoder_heads": 8,
   "depth": 1,
-  "dim": 1024,
-  "dim_head": 64,
+  "dropout": 0.0,
+  "embed_dim": 1024,
   "heads": 8,
-  "input_projection": 512,
+  "input_dim": 512,
+  "lambda_kl": 0.001,
   "masked_only": false,
   "mlp_dim": 2048,
-  "rand_noise_prob": 0.1
-}
+  "num_cls_tokens": 2,
+  "num_modalities": 6,
+  "num_register_tokens": 4,
+  "transformers_version": "5.2.0"
+}