diff --git a/pkgs/development/python-modules/accelerate/default.nix b/pkgs/development/python-modules/accelerate/default.nix index d38a2fe1e37f..e90ed4ddfe38 100644 --- a/pkgs/development/python-modules/accelerate/default.nix +++ b/pkgs/development/python-modules/accelerate/default.nix @@ -25,6 +25,7 @@ evaluate, parameterized, pytestCheckHook, + torchvision, transformers, config, cudatoolkit, @@ -62,6 +63,7 @@ buildPythonPackage (finalAttrs: { evaluate parameterized pytestCheckHook + torchvision transformers writableTmpDirAsHomeHook ];