Add modules.json to start using 1_Pooling/config.json

README.md

@@ -1,5 +1,11 @@
 ---
 license: cc-by-nc-4.0
+pipeline_tag: feature-extraction
+tags:
+- transformers
+- sentence-transformers
+- code
+- retrieval
 ---
 <h1 align="center">Salesforce/SFR-Embedding-Code-400M_R</h1>
 
@@ -64,6 +70,7 @@ embeddings = outputs.last_hidden_state[:, 0]
 embeddings = F.normalize(embeddings, p=2, dim=1)
 scores = (embeddings[:1] @ embeddings[1:].T) * 100
 print("Similarity Scores:", scores.tolist())
+# Similarity Scores: [[74.84745025634766, 65.39266967773438]]
 ```
 
 ### Sentence Transformers
@@ -80,7 +87,9 @@ sentences = [
 
 model = SentenceTransformer('Salesforce/SFR-Embedding-Code-400M_R', trust_remote_code=True)
 embeddings = model.encode(sentences)
-print(cos_sim(embeddings[0], embeddings[1:]))
+similarities = cos_sim(embeddings[0], embeddings[1:])
+print(similarities)
+# tensor([[0.7485, 0.6539]])
 ```
 
 ### Citation

modules.json

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

sentence_bert_config.json

@@ -0,0 +1,4 @@
+{
+  "max_seq_length": 8192,
+  "do_lower_case": false
+}