Fix error when encoding empty set of strings with convert_to_tensor=T…

…rue (#1775) * Fix error when encoding empty set of strings with convert_to_tensor=True * Place common behaviour in if-branch --------- Co-authored-by: Tom Aarsen <[email protected]>
UKPLab · Dec 18, 2023 · 1c396f8 · 1c396f8
1 parent 2a639f0
commit 1c396f8
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/sentence_transformers/SentenceTransformer.py b/sentence_transformers/SentenceTransformer.py
@@ -215,7 +215,10 @@ def encode(self, sentences: Union[str, List[str]],
         all_embeddings = [all_embeddings[idx] for idx in np.argsort(length_sorted_idx)]
 
         if convert_to_tensor:
-            all_embeddings = torch.stack(all_embeddings)
+            if len(all_embeddings):
+                all_embeddings = torch.stack(all_embeddings)
+            else:
+                all_embeddings = torch.Tensor()
         elif convert_to_numpy:
             all_embeddings = np.asarray([emb.numpy() for emb in all_embeddings])