Fix #1670

UKPLab · Aug 15, 2022 · d8982c9 · d8982c9
1 parent 07fd196
commit d8982c9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/sentence_transformers/util.py b/sentence_transformers/util.py
@@ -371,7 +371,7 @@ def community_detection(embeddings, threshold=0.75, min_community_size=10, batch
                 top_val_large, top_idx_large = cos_scores[i].topk(k=sort_max_size, largest=True)
 
                 # Check if we need to increase sort_max_size
-                while top_val_large[-1] > threshold:
+                while top_val_large[-1] > threshold and sort_max_size < len(embeddings):
                     sort_max_size = min(2 * sort_max_size, len(embeddings))
                     top_val_large, top_idx_large = cos_scores[i].topk(k=sort_max_size, largest=True)