Skip to content

Commit

Permalink
Also update return docstring of encode_multi_process (#2548)
Browse files Browse the repository at this point in the history
  • Loading branch information
tomaarsen authored Mar 19, 2024
1 parent caaa28d commit e6af66f
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion sentence_transformers/SentenceTransformer.py
Original file line number Diff line number Diff line change
Expand Up @@ -476,7 +476,7 @@ def encode_multi_process(
:param chunk_size: Sentences are chunked and sent to the individual processes. If none, it determine a sensible size.
:param normalize_embeddings: Whether to normalize returned vectors to have length 1. In that case,
the faster dot-product (util.dot_score) instead of cosine similarity can be used.
:return: Numpy matrix with all embeddings
:return: 2d numpy array with shape [num_inputs, output_dimension]
"""

if chunk_size is None:
Expand Down

0 comments on commit e6af66f

Please sign in to comment.