Update README.md
Browse filesEdits to the matryoshka_dims for truncating the dimensions when loading the embedding model
README.md
CHANGED
|
@@ -119,7 +119,7 @@ Then you can load this model and run inference.
|
|
| 119 |
from sentence_transformers import SentenceTransformer
|
| 120 |
|
| 121 |
# Download from the 🤗 Hub
|
| 122 |
-
matryoshka_dims = [768,512,256,128,64] # for
|
| 123 |
model = SentenceTransformer("DannyAI/embedding_fine_tuning_adaptive_layer_matryoshka2dloss_bge_large_en_v1.5", truncate_dim=matryoshka_dims[0])
|
| 124 |
# Run inference
|
| 125 |
sentences = [
|
|
|
|
| 119 |
from sentence_transformers import SentenceTransformer
|
| 120 |
|
| 121 |
# Download from the 🤗 Hub
|
| 122 |
+
matryoshka_dims = [768,512,256,128,64] # for truncating the dimensions
|
| 123 |
model = SentenceTransformer("DannyAI/embedding_fine_tuning_adaptive_layer_matryoshka2dloss_bge_large_en_v1.5", truncate_dim=matryoshka_dims[0])
|
| 124 |
# Run inference
|
| 125 |
sentences = [
|