diff --git a/setup.py b/setup.py index 8e13fd6840739c..a1e4e7d7184e40 100644 --- a/setup.py +++ b/setup.py @@ -275,7 +275,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"]