PartAI
/

Tooka-SBERT-V2-Small

Sentence Similarity

sentence-transformers

feature-extraction

loss:CachedMultipleNegativesRankingLoss

text-embeddings-inference

Model card Files Files and versions Community

mohalisad commited on May 24

Commit

f096695

·

verified ·

1 Parent(s): 6e5e9d7

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -66,7 +66,7 @@ sentences = [
 ]
 embeddings = model.encode(sentences)
 print(embeddings.shape)
-# [3, 1024]
 # Get the similarity scores for the embeddings
 similarities = model.similarity(embeddings, embeddings)

 ]
 embeddings = model.encode(sentences)
 print(embeddings.shape)
+# [3, 768]
 # Get the similarity scores for the embeddings
 similarities = model.similarity(embeddings, embeddings)