improve error

UKPLab · Jul 5, 2024 · a93071b · a93071b
1 parent c88dd4e
commit a93071b
Showing 1 changed file with 3 additions and 1 deletion.
diff --git 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)]