Add new SentenceTransformer model with an onnx backend
#1
by
gapendl
- opened
- 1_Pooling/config.json +2 -1
- config.json +1 -1
- config_sentence_transformers.json +7 -4
- onnx/model.onnx +3 -0
1_Pooling/config.json
CHANGED
@@ -5,5 +5,6 @@
|
|
5 |
"pooling_mode_max_tokens": false,
|
6 |
"pooling_mode_mean_sqrt_len_tokens": false,
|
7 |
"pooling_mode_weightedmean_tokens": false,
|
8 |
-
"pooling_mode_lasttoken": false
|
|
|
9 |
}
|
|
|
5 |
"pooling_mode_max_tokens": false,
|
6 |
"pooling_mode_mean_sqrt_len_tokens": false,
|
7 |
"pooling_mode_weightedmean_tokens": false,
|
8 |
+
"pooling_mode_lasttoken": false,
|
9 |
+
"include_prompt": true
|
10 |
}
|
config.json
CHANGED
@@ -18,7 +18,7 @@
|
|
18 |
"pad_token_id": 0,
|
19 |
"position_embedding_type": "absolute",
|
20 |
"torch_dtype": "float32",
|
21 |
-
"transformers_version": "4.
|
22 |
"type_vocab_size": 2,
|
23 |
"use_cache": true,
|
24 |
"vocab_size": 31102
|
|
|
18 |
"pad_token_id": 0,
|
19 |
"position_embedding_type": "absolute",
|
20 |
"torch_dtype": "float32",
|
21 |
+
"transformers_version": "4.45.2",
|
22 |
"type_vocab_size": 2,
|
23 |
"use_cache": true,
|
24 |
"vocab_size": 31102
|
config_sentence_transformers.json
CHANGED
@@ -1,7 +1,10 @@
|
|
1 |
{
|
2 |
"__version__": {
|
3 |
-
"sentence_transformers": "2.
|
4 |
-
"transformers": "4.
|
5 |
-
"pytorch": "2.
|
6 |
-
}
|
|
|
|
|
|
|
7 |
}
|
|
|
1 |
{
|
2 |
"__version__": {
|
3 |
+
"sentence_transformers": "3.2.1",
|
4 |
+
"transformers": "4.45.2",
|
5 |
+
"pytorch": "2.5.0+cu124"
|
6 |
+
},
|
7 |
+
"prompts": {},
|
8 |
+
"default_prompt_name": null,
|
9 |
+
"similarity_fn_name": null
|
10 |
}
|
onnx/model.onnx
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:c07f54705d876a10b2c7ab6346bbdc2c111db311bb0ba5e2038d55dbda152f6b
|
3 |
+
size 437626376
|