Add Model with custom pooling layer

1_WKPooling/config.json

@@ -0,0 +1,5 @@
+{
+  "word_embedding_dimension": 768,
+  "layer_start": 4,
+  "context_window_size": 2
+}

config.json

@@ -0,0 +1,57 @@
+{
+  "_num_labels": 2,
+  "activation": "gelu",
+  "architectures": [
+    "DistilBertModel"
+  ],
+  "attention_dropout": 0.1,
+  "bad_words_ids": null,
+  "bos_token_id": null,
+  "decoder_start_token_id": null,
+  "dim": 768,
+  "do_sample": false,
+  "dropout": 0.1,
+  "early_stopping": false,
+  "eos_token_id": null,
+  "finetuning_task": null,
+  "hidden_dim": 3072,
+  "id2label": {
+    "0": "LABEL_0",
+    "1": "LABEL_1"
+  },
+  "initializer_range": 0.02,
+  "is_decoder": false,
+  "is_encoder_decoder": false,
+  "label2id": {
+    "LABEL_0": 0,
+    "LABEL_1": 1
+  },
+  "length_penalty": 1.0,
+  "max_length": 20,
+  "max_position_embeddings": 512,
+  "min_length": 0,
+  "model_type": "distilbert",
+  "n_heads": 12,
+  "n_layers": 6,
+  "no_repeat_ngram_size": 0,
+  "num_beams": 1,
+  "num_return_sequences": 1,
+  "output_attentions": false,
+  "output_hidden_states": true,
+  "output_past": true,
+  "pad_token_id": 0,
+  "prefix": null,
+  "pruned_heads": {},
+  "qa_dropout": 0.1,
+  "repetition_penalty": 1.0,
+  "seq_classif_dropout": 0.2,
+  "sinusoidal_pos_embds": false,
+  "task_specific_params": null,
+  "temperature": 1.0,
+  "tie_weights_": true,
+  "top_k": 50,
+  "top_p": 1.0,
+  "torchscript": false,
+  "use_bfloat16": false,
+  "vocab_size": 30522
+}

modules.json

@@ -0,0 +1,14 @@
+[
+  {
+    "idx": 0,
+    "name": "0",
+    "path": "",
+    "type": "sentence_transformers.models.Transformer"
+  },
+  {
+    "idx": 1,
+    "name": "1",
+    "path": "1_WKPooling",
+    "type": "sentence_transformers.models.WKPooling"
+  }
+]

pytorch_model.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3e9f8b1e85657b209941057dd15542e5b85d18f110ad5edf8500163dec9bf8c1
+size 265472230

sentence_bert_config.json

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

special_tokens_map.json

@@ -0,0 +1 @@
+{"unk_token": "[UNK]", "sep_token": "[SEP]", "pad_token": "[PAD]", "cls_token": "[CLS]", "mask_token": "[MASK]"}

tokenizer_config.json

@@ -0,0 +1 @@
+{"do_lower_case": true, "max_len": 512, "unk_token": "[UNK]", "sep_token": "[SEP]", "pad_token": "[PAD]", "cls_token": "[CLS]", "mask_token": "[MASK]"}