Mimic SentenceTransformer

1/config.json

@@ -1,3 +0,0 @@
-{
-  "pooling_mode": "mean"
-}

1_Pooling/config.json

@@ -0,0 +1,7 @@
+{
+  "word_embedding_dimension": 768,
+  "pooling_mode_mean_tokens": true,
+  "pooling_mode_cls_token": false,
+  "pooling_mode_max_tokens": false,
+  "pooling_mode_mean_sqrt_len_tokens": false
+}

2/config.json

@@ -1,4 +0,0 @@
-{
-  "d_from": 768,
-  "d_to": 3072
-}

2_Dense/config.json

@@ -0,0 +1,6 @@
+{
+  "in_features": 768,
+  "out_features": 3072,
+  "bias": false,
+  "activation_function": "torch.nn.modules.linear.Identity"
+}

README.md

@@ -0,0 +1,8 @@
+---
+pipeline_tag: sentence-similarity
+tags:
+  - sentence-transformers
+  - feature-extraction
+  - sentence-similarity
+  - transformers
+---

config_sentence_transformers.json

@@ -0,0 +1,7 @@
+{
+  "__version__": {
+    "sentence_transformers": "2.0.0",
+    "transformers": "4.7.0",
+    "pytorch": "1.9.0+cu102"
+  }
+}

modules.json

@@ -0,0 +1,20 @@
+[
+  {
+    "idx": 0,
+    "name": "0",
+    "path": "",
+    "type": "sentence_transformers.models.Transformer"
+  },
+  {
+    "idx": 1,
+    "name": "1",
+    "path": "1_Pooling",
+    "type": "sentence_transformers.models.Pooling"
+  },
+  {
+    "idx": 2,
+    "name": "2",
+    "path": "2_Dense",
+    "type": "sentence_transformers.models.Dense"
+  }
+]

sentence_bert_config.json

@@ -0,0 +1,3 @@
+{
+  "max_seq_length": 256
+}