diff options
Diffstat (limited to 'pkgs/development/python-modules/optimum/default.nix')
-rw-r--r-- | pkgs/development/python-modules/optimum/default.nix | 34 |
1 files changed, 21 insertions, 13 deletions
diff --git a/pkgs/development/python-modules/optimum/default.nix b/pkgs/development/python-modules/optimum/default.nix index 07c80298754a7..d23fe4ab529b4 100644 --- a/pkgs/development/python-modules/optimum/default.nix +++ b/pkgs/development/python-modules/optimum/default.nix @@ -3,29 +3,35 @@ buildPythonPackage, fetchFromGitHub, pythonOlder, + + # build-system + setuptools, + + # dependencies coloredlogs, datasets, + huggingface-hub, + numpy, + packaging, + sympy, + torch, + transformers, + + # optional-dependencies diffusers, evaluate, h5py, - huggingface-hub, - numpy, onnx, onnxruntime, - packaging, protobuf, - setuptools, - sympy, tensorflow, tf2onnx, timm, - torch, - transformers, }: buildPythonPackage rec { pname = "optimum"; - version = "1.20.0"; + version = "1.22.0"; pyproject = true; disabled = pythonOlder "3.7"; @@ -34,11 +40,13 @@ buildPythonPackage rec { owner = "huggingface"; repo = "optimum"; rev = "refs/tags/v${version}"; - hash = "sha256-aQNDVNWTgY2LEtug229SEZRMvKHpsQfiTPWW4Lh3hs4="; + hash = "sha256-fVpSoa6s7puYO8BiA6aSAhSOTfiZJgaX5iwYkd9VdVo="; }; build-system = [ setuptools ]; + pythonRelaxDeps = [ "transformers" ]; + dependencies = [ coloredlogs datasets @@ -108,12 +116,12 @@ buildPythonPackage rec { pythonImportsCheck = [ "optimum" ]; - meta = with lib; { + meta = { description = "Accelerate training and inference of 🤗 Transformers and 🤗 Diffusers with easy to use hardware optimization tools"; mainProgram = "optimum-cli"; homepage = "https://github.com/huggingface/optimum"; - changelog = "https://github.com/huggingface/optimum/releases/tag/${src.rev}"; - license = licenses.asl20; - maintainers = with maintainers; [ natsukium ]; + changelog = "https://github.com/huggingface/optimum/releases/tag/${lib.removePrefix "refs/tags/" src.rev}"; + license = lib.licenses.asl20; + maintainers = with lib.maintainers; [ natsukium ]; }; } |