diff --git a/setup.py b/setup.py index 64402852a4..ed785fe5f5 100644 --- a/setup.py +++ b/setup.py @@ -152,9 +152,7 @@ def _parse_requirements_file(file_path): "scikit-learn", "seqeval", ] -_sentence_transformers_integration_deps = [ - "optimum-deepsparse @ git://github.com/neuralmagic/optimum-deepsparse.git" -] + _torch_deps +_sentence_transformers_integration_deps = ["optimum-deepsparse"] + _torch_deps # haystack dependencies are installed from a requirements file to avoid # conflicting versions with NM's deepsparse/transformers