diff --git a/setup.py b/setup.py index e2311b13a25e..20c9ea61f5f2 100644 --- a/setup.py +++ b/setup.py @@ -92,7 +92,6 @@ "jaxlib>=0.1.65,<=0.3.6", "modelcards>=0.1.4", "numpy", - "onnxruntime", "onnxruntime-gpu", "pytest", "pytest-timeout", @@ -179,7 +178,6 @@ def run(self): extras["training"] = deps_list("accelerate", "datasets", "tensorboard", "modelcards") extras["test"] = deps_list( "datasets", - "onnxruntime", "onnxruntime-gpu", "pytest", "pytest-timeout",