diff --git a/setup.py b/setup.py index 669384667ad0dc..5b6a461783cc54 100644 --- a/setup.py +++ b/setup.py @@ -273,7 +273,6 @@ def run(self): extras["flax"] = deps_list("jax", "jaxlib", "flax", "optax") extras["tokenizers"] = deps_list("tokenizers") -extras["safetensors"] = deps_list("safetensors") extras["ftfy"] = deps_list("ftfy") extras["onnxruntime"] = deps_list("onnxruntime", "onnxruntime-tools") extras["onnx"] = deps_list("onnxconverter-common", "tf2onnx") + extras["onnxruntime"]