Something went wrong. Try again.
What happens if I push all of nixpkgs
Something went wrong. Try again.
2.3 kB · 115 lines
Nix
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116{ lib, buildPythonPackage, fetchFromGitHub, pythonOlder, coloredlogs, datasets, evaluate, h5py, huggingface-hub, numpy, onnx, onnxruntime, packaging, protobuf, sympy, tensorflow, tf2onnx, timm, torch, transformers}:
buildPythonPackage rec { pname = "optimum"; version = "1.16.2"; format = "setuptools";
disabled = pythonOlder "3.7";
src = fetchFromGitHub { owner = "huggingface"; repo = "optimum"; rev = "refs/tags/v${version}"; hash = "sha256-bEwCUPZJT1UTD8mQJKg+Nwag6kpn4076uyKocKI/+/M="; };
propagatedBuildInputs = [ coloredlogs datasets huggingface-hub numpy packaging sympy torch transformers ] ++ transformers.optional-dependencies.sentencepiece;
passthru.optional-dependencies = { onnxruntime = [ onnx onnxruntime datasets evaluate protobuf ]; exporters = [ onnx onnxruntime timm ]; exporters-tf = [ tensorflow tf2onnx onnx onnxruntime timm h5py numpy ]; diffusers = [ # diffusers ]; intel = [ # optimum-intel ]; openvino = [ # optimum-intel ]; # ++ optimum-intel.optional-dependencies.openvino; nncf = [ # optimum-intel ]; # ++ optimum-intel.optional-dependencies.nncf; neural-compressor = [ # optimum-intel ]; # ++ optimum-intel.optional-dependencies.neural-compressor; graphcore = [ # optimum-graphcore ]; habana = [ transformers # optimum-habana ]; neuron = [ # optimum-neuron ]; # ++ optimum-neuron.optional-dependencies.neuron; neuronx = [ # optimum-neuron ]; # ++ optimum-neuron.optional-dependencies.neuronx; furiosa = [ # optimum-furiosa ]; };
# almost all tests try to connect to https://huggingface.co doCheck = false;
pythonImportsCheck = [ "optimum" ];
meta = with lib; { description = "Accelerate training and inference of 🤗 Transformers and 🤗 Diffusers with easy to use hardware optimization tools"; homepage = "https://github.com/huggingface/optimum"; changelog = "https://github.com/huggingface/optimum/releases/tag/${src.rev}"; license = licenses.asl20; maintainers = with maintainers; [ natsukium ]; };}