From ab85d71a5a6df6054f88c9cabfbf2e4ef8d0284d Mon Sep 17 00:00:00 2001
From: alperctnkaya <33162362+alperctnkaya@users.noreply.github.com>
Date: Wed, 2 Oct 2024 09:36:00 +0300
Subject: [PATCH] multi-GPU support for mine_hard_negatives

---
 sentence_transformers/util.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/sentence_transformers/util.py b/sentence_transformers/util.py
index ee5f1a00a..be523dde1 100644
--- a/sentence_transformers/util.py
+++ b/sentence_transformers/util.py
@@ -533,7 +533,7 @@ def mine_hard_negatives(
     faiss_batch_size: int = 16384,
     use_faiss: bool = False,
     verbose: bool = True,
-    use_multiple_gpus = True,
+    use_multiple_gpus = False,
 ) -> Dataset:
     """
     Add hard negatives to a dataset of (anchor, positive) pairs to create (anchor, positive, negative) triplets or