Sentence Similarity
sentence-transformers
ONNX
Safetensors
Transformers
bert
feature-extraction
text-embeddings-inference
Instructions to use kurier-digital/geronimo-base with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- sentence-transformers
How to use kurier-digital/geronimo-base with sentence-transformers:
from sentence_transformers import SentenceTransformer model = SentenceTransformer("kurier-digital/geronimo-base") sentences = [ "That is a happy person", "That is a happy dog", "That is a very happy person", "Today is a sunny day" ] embeddings = model.encode(sentences) similarities = model.similarity(embeddings, embeddings) print(similarities.shape) # [4, 4] - Transformers
How to use kurier-digital/geronimo-base with Transformers:
# pip install -U transformers accelerate # Load model directly from transformers import AutoTokenizer, AutoModel tokenizer = AutoTokenizer.from_pretrained("kurier-digital/geronimo-base") model = AutoModel.from_pretrained("kurier-digital/geronimo-base", device_map="auto") - Notebooks
- Google Colab
- Kaggle
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": "
|
| 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
|