Sparse Encoder Update

3_CSRSparsity/pytorch_model.bin → 3_SparseAutoEncoder/model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:d42f54bc215c6f95c64b9200bde13d2908d97c4af0f6e874fd3d84a75c2823b3
-size 268650784
+oid sha256:9a6471d3889c0828dbdbd53c2eabef4027554d0ac705c029496358135f94b303
+size 268648800

config_sentence_transformers.json

@@ -1,27 +1,13 @@
 {
   "__version__": {
-    "sentence_transformers": "4.2.0",
+    "sentence_transformers": "5.0.0",
     "transformers": "4.47.0",
     "pytorch": "2.5.1+cu12"
   },
   "prompts": {
-    "Banking77Classification": "Instruct: Given a question, please describe the intent of this question. \n Question: ",
-    "MTOPIntentClassification": "Instruct: Given a question, please describe the intent of this question. \n Question: ",
-    "TweetSentimentClassification": "Classify the sentiment of a given tweet as either positive, negative, or neutral.",
-    "BiorxivClusteringP2P.v2": "Identify the main category of Biorxiv papers based on the titles and abstracts",
-    "BiorxivClusteringS2S.v2": "Identify the main category of Biorxiv papers based on the titles",
-    "TwentyNewsgroupsClustering": "Identify the topic or theme of the given news articles",
-    "FiQA2018": {
-      "query": "Given a financial question, retrieve relevant passages that answer the query"
-    },
-    "SciFact": {
-      "query": "Given a scientific claim, retrieve documents that support or refute the claim"
-    },
-    "NFCorpus": {
-      "query": "Given a question, retrieve relevant documents that answer the question"
-    }
+    "Banking77Classification": "Instruct: Given a question, please describe the intent of this question. \n Question: "
   },
   "default_prompt_name": null,
   "model_type": "SparseEncoder",
-  "similarity_fn_name": "dot"
+  "similarity_fn_name": "cosine"
 }

modules.json

@@ -20,7 +20,7 @@
   {
     "idx": 3,
     "name": "3",
-    "path": "3_CSRSparsity",
-    "type": "sentence_transformers.sparse_encoder.models.CSRSparsity"
+    "path": "3_SparseAutoEncoder",
+    "type": "sentence_transformers.sparse_encoder.models.SparseAutoEncoder"
   }
 ]