mixedbread-ai
/

mxbai-embed-large-v1

Feature Extraction

sentence-transformers

Transformers.js

text-embeddings-inference

Inference Endpoints

Model card Files Files and versions Community

aamirshakir commited on Aug 5

Commit

e4614dd

•

1 Parent(s): 2a9c989

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -2684,7 +2684,7 @@ def pooling(outputs: torch.Tensor, inputs: Dict,  strategy: str = 'cls') -> np.n
         outputs = outputs[:, 0]
     elif strategy == 'mean':
         outputs = torch.sum(
-            outputs * inputs["attention_mask"][:, :, None], dim=1) / torch.sum(inputs["attention_mask"])
     else:
         raise NotImplementedError
     return outputs.detach().cpu().numpy()

         outputs = outputs[:, 0]
     elif strategy == 'mean':
         outputs = torch.sum(
+            outputs * inputs["attention_mask"][:, :, None], dim=1) / torch.sum(inputs["attention_mask"], dim=1, keepdim=True)
     else:
         raise NotImplementedError
     return outputs.detach().cpu().numpy()