diff --git a/pkgs/development/python-modules/optimum/default.nix b/pkgs/development/python-modules/optimum/default.nix index dd5598b2d808..3ff806860f45 100644 --- a/pkgs/development/python-modules/optimum/default.nix +++ b/pkgs/development/python-modules/optimum/default.nix @@ -17,7 +17,6 @@ # optional-dependencies diffusers, - evaluate, h5py, onnx, onnxruntime, @@ -56,24 +55,25 @@ buildPythonPackage rec { optional-dependencies = { onnxruntime = [ onnx - onnxruntime datasets - evaluate protobuf + onnxruntime ]; exporters = [ onnx - onnxruntime timm + onnxruntime + protobuf ]; exporters-tf = [ - tensorflow - tf2onnx onnx - onnxruntime timm h5py + tf2onnx + onnxruntime numpy + datasets + tensorflow ]; diffusers = [ diffusers ]; intel = [