fix example

Signed-off-by: Feng, Jiqing <[email protected]>
UKPLab · Nov 26, 2024 · c6a2cd2 · c6a2cd2
1 parent 0da40fa
commit c6a2cd2
Showing 1 changed file with 0 additions and 3 deletions.
diff --git a/examples/evaluation/evaluation_inference_speed.py b/examples/evaluation/evaluation_inference_speed.py
@@ -15,9 +15,6 @@
 
 from sentence_transformers import SentenceTransformer
 
-# Limit torch to 4 threads
-torch.set_num_threads(4)
-
 
 model_name = sys.argv[1] if len(sys.argv) > 1 else "bert-base-nli-mean-tokens"