We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 0d609d0 commit dc02ea1Copy full SHA for dc02ea1
fastembed/text/pooled_embedding.py
@@ -98,8 +98,6 @@ def _get_worker_class(cls) -> Type[OnnxTextEmbeddingWorker]:
98
def mean_pooling(
99
cls, model_output: NumpyArray, attention_mask: NDArray[np.int64]
100
) -> NumpyArray:
101
- # token_embeddings = model_output.astype(np.float32)
102
- # attention_mask = attention_mask.astype(np.float32)
103
return mean_pooling(model_output, attention_mask)
104
105
@classmethod
0 commit comments