updates

README.md

@@ -4,7 +4,6 @@ tags:
 - sentence-transformers
 - feature-extraction
 - sentence-similarity
-- transformers
 ---
 
 # {MODEL_NAME}
@@ -34,44 +33,6 @@ print(embeddings)
 
 
 
-## Usage (HuggingFace Transformers)
-Without [sentence-transformers](https://www.SBERT.net), you can use the model like this: First, you pass your input through the transformer model, then you have to apply the right pooling-operation on-top of the contextualized word embeddings.
-
-```python
-from transformers import AutoTokenizer, AutoModel
-import torch
-
-
-#Mean Pooling - Take attention mask into account for correct averaging
-def mean_pooling(model_output, attention_mask):
-    token_embeddings = model_output[0] #First element of model_output contains all token embeddings
-    input_mask_expanded = attention_mask.unsqueeze(-1).expand(token_embeddings.size()).float()
-    return torch.sum(token_embeddings * input_mask_expanded, 1) / torch.clamp(input_mask_expanded.sum(1), min=1e-9)
-
-
-# Sentences we want sentence embeddings for
-sentences = ['This is an example sentence', 'Each sentence is converted']
-
-# Load model from HuggingFace Hub
-tokenizer = AutoTokenizer.from_pretrained('{MODEL_NAME}')
-model = AutoModel.from_pretrained('{MODEL_NAME}')
-
-# Tokenize sentences
-encoded_input = tokenizer(sentences, padding=True, truncation=True, return_tensors='pt')
-
-# Compute token embeddings
-with torch.no_grad():
-    model_output = model(**encoded_input)
-
-# Perform pooling. In this case, mean pooling.
-sentence_embeddings = mean_pooling(model_output, encoded_input['attention_mask'])
-
-print("Sentence embeddings:")
-print(sentence_embeddings)
-```
-
-
-
 ## Evaluation Results
 
 <!--- Describe how your model was evaluated -->
@@ -115,8 +76,9 @@ Parameters of the fit()-Method:
 ## Full Model Architecture
 ```
 SentenceTransformer(
-  (0): Transformer({'max_seq_length': 514, 'do_lower_case': False}) with Transformer model: MPNetModel 
+  (0): Transformer({'max_seq_length': 384, 'do_lower_case': False}) with Transformer model: MPNetModel 
   (1): Pooling({'word_embedding_dimension': 768, 'pooling_mode_cls_token': False, 'pooling_mode_mean_tokens': True, 'pooling_mode_max_tokens': False, 'pooling_mode_mean_sqrt_len_tokens': False})
+  (2): Normalize()
 )
 ```
 

config.json

@@ -1,5 +1,5 @@
 {
-  "_name_or_path": "/root/.cache/torch/sentence_transformers/microsoft_mpnet-base",
+  "_name_or_path": "/root/.cache/torch/sentence_transformers/sentence-transformers_all-mpnet-base-v2/",
   "architectures": [
     "MPNetModel"
   ],

config_sentence_transformers.json

@@ -1,7 +1,7 @@
 {
   "__version__": {
-    "sentence_transformers": "2.1.0",
-    "transformers": "4.21.1",
-    "pytorch": "1.7.0+cu110"
+    "sentence_transformers": "2.0.0",
+    "transformers": "4.6.1",
+    "pytorch": "1.8.1"
   }
 }

modules.json

@@ -10,5 +10,11 @@
     "name": "1",
     "path": "1_Pooling",
     "type": "sentence_transformers.models.Pooling"
+  },
+  {
+    "idx": 2,
+    "name": "2",
+    "path": "2_Normalize",
+    "type": "sentence_transformers.models.Normalize"
   }
 ]

pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:35de644b5509f5efa3a0365d02ef2f48ff01e1284eb0bb80ff6e0d85680e02f6
+oid sha256:2041457ed41c970a055ed9202a53a80dda4ba3ae8e3d9b31d780baea927ba402
 size 438019895

sentence_bert_config.json

@@ -1,4 +1,4 @@
 {
-  "max_seq_length": 514,
+  "max_seq_length": 384,
   "do_lower_case": false
 }

tokenizer.json

@@ -2,7 +2,7 @@
   "version": "1.0",
   "truncation": {
     "direction": "Right",
-    "max_length": 514,
+    "max_length": 384,
     "strategy": "LongestFirst",
     "stride": 0
   },

tokenizer_config.json

@@ -4,7 +4,8 @@
   "do_lower_case": true,
   "eos_token": "</s>",
   "mask_token": "<mask>",
-  "name_or_path": "/root/.cache/torch/sentence_transformers/microsoft_mpnet-base",
+  "model_max_length": 512,
+  "name_or_path": "/root/.cache/torch/sentence_transformers/sentence-transformers_all-mpnet-base-v2/",
   "pad_token": "<pad>",
   "sep_token": "</s>",
   "special_tokens_map_file": null,