From c2509dc9db837388da9664479b824f2dc08ea3b6 Mon Sep 17 00:00:00 2001 From: Weijia Wang <9713184+wegank@users.noreply.github.com> Date: Sun, 25 May 2025 12:49:53 +0200 Subject: [PATCH] python3Packages.optimum: update optional dependencies --- .../development/python-modules/optimum/default.nix | 14 +++++++------- 1 file changed, 7 insertions(+), 7 deletions(-) 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 = [