Upload torch_config.json

torch_config.json

@@ -0,0 +1,30 @@
+{
+    "text_encoder": {
+        "model_type": "bert",
+        "dim": 384,
+        "context_dim": 768,
+        "vocab_size": 250037,
+        "padding_idx": 0,
+        "num_layers": 12,
+        "num_heads": 12,
+        "embedding_dim": 256,
+        "multimodal_layers_ids": [8, 9, 10, 11],
+        "head_one_neuron": true,
+        "pooling": "mean",
+        "max_position_embeddings": 77,
+        "dropout_prob": 0.1
+    },
+    "image_encoder": {
+        "dim": 768,
+        "patch_size": 16,
+        "image_size": 224,
+        "num_layers": 12,
+        "num_heads": 12,
+        "embedding_dim": 256,
+        "pooling": "cls"
+    },
+    "preprocess": {
+        "image_size": 224,
+        "tokenizer": "sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2"
+    }
+}