diff --git a/sentence_transformers/SentenceTransformer.py b/sentence_transformers/SentenceTransformer.py index fe808a314..070ccabe5 100644 --- a/sentence_transformers/SentenceTransformer.py +++ b/sentence_transformers/SentenceTransformer.py @@ -615,6 +615,9 @@ def encode( sentences = [sentences] input_was_string = True + if not isinstance(sentences, (list, np.ndarray)): + sentences = list(sentences) + if prompt is None: if prompt_name is not None: try: