diff --git a/fastembed/sparse/bm25.py b/fastembed/sparse/bm25.py index 2b124070..e095c627 100644 --- a/fastembed/sparse/bm25.py +++ b/fastembed/sparse/bm25.py @@ -375,7 +375,7 @@ def query_embed(self, query: str | Iterable[str], **kwargs: Any) -> Iterable[Spa stemmed_tokens = self._stem(tokens) token_ids = np.array( list(set(self.compute_token_id(token) for token in stemmed_tokens)), - dtype=np.int32, + dtype=np.int64, ) values = np.ones_like(token_ids) yield SparseEmbedding(indices=token_ids, values=values)