diff --git a/setup.py b/setup.py index 15dc288d2572d7..dae1f0eca41b49 100644 --- a/setup.py +++ b/setup.py @@ -288,7 +288,6 @@ def run(self): extras["onnxruntime"] = deps_list("onnxruntime", "onnxruntime-tools") extras["onnx"] = deps_list("onnxconverter-common", "tf2onnx") + extras["onnxruntime"] extras["modelcreation"] = deps_list("cookiecutter") - extras["sagemaker"] = deps_list("sagemaker") extras["deepspeed"] = deps_list("deepspeed") + extras["accelerate"] extras["optuna"] = deps_list("optuna")