diff --git a/examples/evaluation/evaluation_inference_speed.py b/examples/evaluation/evaluation_inference_speed.py index fb3c8d909..837c7ddea 100644 --- 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"