mirrors/nixpkgs
1
0
Fork 1
mirror of https://github.com/NixOS/nixpkgs.git synced 2024-11-19 12:11:28 +00:00
Code Issues Wiki Activity

python311Packages.optimum: 1.19.2 -> 1.20.0

Browse source
This commit is contained in:
R. Ryantm 2024-06-01 18:39:31 +00:00 committed by natsukium
parent 61c1d28215
commit 3542c02fa8
No known key found for this signature in database
GPG key ID: 9EA45A31DB994C53
1 changed files with 10 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

18
pkgs/development/python-modules/optimum/default.nix
View file

@ -5,6 +5,7 @@
pythonOlder,
coloredlogs,
datasets,
diffusers,
evaluate,
h5py,
huggingface-hub,
@ -13,6 +14,7 @@
onnxruntime,
packaging,
protobuf,
setuptools,
sympy,
tensorflow,
tf2onnx,
@ -23,8 +25,8 @@
buildPythonPackage rec {
pname = "optimum";
version = "1.19.2";
format = "setuptools";
version = "1.20.0";
pyproject = true;
disabled = pythonOlder "3.7";
@ -32,10 +34,12 @@ buildPythonPackage rec {
owner = "huggingface";
repo = "optimum";
rev = "refs/tags/v${version}";
hash = "sha256-GYgLh6vlOoVvTvhNPfBT4YEqRhB7gZMqlOuab6maVGU=";
hash = "sha256-aQNDVNWTgY2LEtug229SEZRMvKHpsQfiTPWW4Lh3hs4=";
};
propagatedBuildInputs = [
build-system = [ setuptools ];
dependencies = [
coloredlogs
datasets
huggingface-hub
@ -46,7 +50,7 @@ buildPythonPackage rec {
transformers
] ++ transformers.optional-dependencies.sentencepiece;
passthru.optional-dependencies = {
optional-dependencies = {
onnxruntime = [
onnx
onnxruntime
@ -68,9 +72,7 @@ buildPythonPackage rec {
h5py
numpy
];
diffusers = [
# diffusers
];
diffusers = [ diffusers ];
intel = [
# optimum-intel
];