diff --git a/pkgs/development/python-modules/optimum/default.nix b/pkgs/development/python-modules/optimum/default.nix index 7238e689b6d3..43925f8ee7ab 100644 --- a/pkgs/development/python-modules/optimum/default.nix +++ b/pkgs/development/python-modules/optimum/default.nix @@ -8,12 +8,10 @@ setuptools, # dependencies - coloredlogs, datasets, huggingface-hub, numpy, packaging, - sympy, torch, transformers, @@ -34,7 +32,7 @@ buildPythonPackage rec { version = "1.24.0"; pyproject = true; - disabled = pythonOlder "3.7"; + disabled = pythonOlder "3.9"; src = fetchFromGitHub { owner = "huggingface"; @@ -48,12 +46,9 @@ buildPythonPackage rec { pythonRelaxDeps = [ "transformers" ]; dependencies = [ - coloredlogs - datasets huggingface-hub numpy packaging - sympy torch transformers ] ++ transformers.optional-dependencies.sentencepiece;