Commit c2509dc9 authored by Weijia Wang's avatar Weijia Wang
Browse files

python3Packages.optimum: update optional dependencies

parent ae1c56ad

pkgs/development/python-modules/optimum/default.nix

+7 −7
Original line number Diff line number Diff line
@@ -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 = [