From a4577cadb163547a9e72d101b992d32496aeea52 Mon Sep 17 00:00:00 2001 From: Ita Zaporozhets Date: Thu, 8 Aug 2024 15:09:59 +0200 Subject: [PATCH] [build-ci-image] --- setup.py | 1 - 1 file changed, 1 deletion(-) 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")