Upload folder using huggingface_hub

1_Pooling/config.json

@@ -1,10 +1,5 @@
 {
-    "word_embedding_dimension": 384,
-    "pooling_mode_cls_token": false,
-    "pooling_mode_mean_tokens": true,
-    "pooling_mode_max_tokens": false,
-    "pooling_mode_mean_sqrt_len_tokens": false,
-    "pooling_mode_weightedmean_tokens": false,
-    "pooling_mode_lasttoken": false,
+    "embedding_dimension": 384,
+    "pooling_mode": "mean",
     "include_prompt": true
 }

README.md

@@ -24,7 +24,7 @@ from sentence_transformers import SentenceTransformer
 model = SentenceTransformer(
     "jrc2139/e5-small-v2-ONNX",
     backend="onnx",
-    model_kwargs={"file_name": "onnx/model_qint8_arm64.onnx"},
+    model_kwargs={"file_name": "onnx/model_q4.onnx"},
     trust_remote_code=True
 )
 ```

config.json

@@ -4,6 +4,7 @@
   ],
   "attention_probs_dropout_prob": 0.1,
   "classifier_dropout": null,
+  "dtype": "float32",
   "hidden_act": "gelu",
   "hidden_dropout_prob": 0.1,
   "hidden_size": 384,
@@ -16,8 +17,7 @@
   "num_hidden_layers": 12,
   "pad_token_id": 0,
   "position_embedding_type": "absolute",
-  "torch_dtype": "float32",
-  "transformers_version": "4.55.4",
+  "transformers_version": "4.57.6",
   "type_vocab_size": 2,
   "use_cache": true,
   "vocab_size": 30522

config_sentence_transformers.json

@@ -1,14 +1,14 @@
 {
-  "model_type": "SentenceTransformer",
   "__version__": {
-    "sentence_transformers": "5.2.2",
-    "transformers": "4.55.4",
-    "pytorch": "2.9.1"
+    "pytorch": "2.11.0",
+    "sentence_transformers": "5.4.1",
+    "transformers": "4.57.6"
   },
+  "default_prompt_name": null,
+  "model_type": "SentenceTransformer",
   "prompts": {
-    "query": "",
-    "document": ""
+    "document": "",
+    "query": ""
   },
-  "default_prompt_name": null,
   "similarity_fn_name": "cosine"
 }

modules.json

@@ -3,18 +3,18 @@
     "idx": 0,
     "name": "0",
     "path": "",
-    "type": "sentence_transformers.models.Transformer"
+    "type": "sentence_transformers.base.modules.transformer.Transformer"
   },
   {
     "idx": 1,
     "name": "1",
     "path": "1_Pooling",
-    "type": "sentence_transformers.models.Pooling"
+    "type": "sentence_transformers.sentence_transformer.modules.pooling.Pooling"
   },
   {
     "idx": 2,
     "name": "2",
     "path": "2_Normalize",
-    "type": "sentence_transformers.models.Normalize"
+    "type": "sentence_transformers.sentence_transformer.modules.normalize.Normalize"
   }
 ]

onnx/model.onnx

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:4e28c35ff4561db572916a56edccda652182e4e49c36ff14362648c7a9ba4c6d
-size 133041944
+oid sha256:ed73190ada37b65a103b2188897602bce416e51dadd4c8402bb93252fac23e4f
+size 133041945

sentence_bert_config.json

@@ -1,4 +1,10 @@
 {
-    "max_seq_length": 512,
-    "do_lower_case": false
+    "transformer_task": "feature-extraction",
+    "modality_config": {
+        "text": {
+            "method": "forward",
+            "method_output_name": "last_hidden_state"
+        }
+    },
+    "module_output_name": "token_embeddings"
 }