diff --git a/sentence_transformers/SentenceTransformer.py b/sentence_transformers/SentenceTransformer.py index 54538aac4..412dd253f 100644 --- a/sentence_transformers/SentenceTransformer.py +++ b/sentence_transformers/SentenceTransformer.py @@ -572,7 +572,9 @@ def encode( elif not output_value: all_embeddings.extend(self._process_all_outputs(out_features)) else: - raise ValueError(f"Got unexpected value for 'output_value' : {output_value}") + raise ValueError( + f"Got unexpected value for 'output_value' : {output_value}. Valid values are 'token_embeddings', 'sentence_embedding' or None." + ) all_embeddings = [all_embeddings[idx] for idx in np.argsort(length_sorted_idx)]