intfloat
/

e5-large-v2

Sentence Similarity

sentence-transformers

Sentence Transformers

Eval Results (legacy)

text-embeddings-inference

Model card Files Files and versions

ababeal commited on Oct 13, 2023

Commit

df40310

·

1 Parent(s): 4aadff9

bug: fixed incorrect variable

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -26,4 +26,4 @@ class EndpointHandler():
         embeddings = average_pool(outputs.last_hidden_state, batch_dict['attention_mask'])
         embeddings = F.normalize(embeddings, p=2, dim=1)
         scores = (embeddings[:2] @ embeddings[2:].T) * 100
-        return embeddings.tolist()

         embeddings = average_pool(outputs.last_hidden_state, batch_dict['attention_mask'])
         embeddings = F.normalize(embeddings, p=2, dim=1)
         scores = (embeddings[:2] @ embeddings[2:].T) * 100
+        return scores.tolist()