fresha
/

e5-large-v2-endpoint

Feature Extraction

Eval Results (legacy)

text-embeddings-inference

Model card Files Files and versions

Hans Elias J commited on Jun 28, 2023

Commit

ce64906

·

1 Parent(s): 0996185

fix norm

Files changed (1) hide show

handler.py +2 -2

handler.py CHANGED Viewed

@@ -30,7 +30,7 @@ class EndpointHandler():
         outputs = self.model(**batch_dict)
-        embeddings = average_pool(outputs.last_hidden_state, batch_dict['attention_mask']).tolist()
-        embeddings = F.normalize(embeddings, p=2, dim=1)
         return embeddings

         outputs = self.model(**batch_dict)
+        embeddings = average_pool(outputs.last_hidden_state, batch_dict['attention_mask'])
+        embeddings = F.normalize(embeddings, p=2, dim=1).tolist()
         return embeddings