diff options
Diffstat (limited to 'Omni/Bild/Deps/accelerate.nix')
-rw-r--r-- | Omni/Bild/Deps/accelerate.nix | 13 |
1 files changed, 9 insertions, 4 deletions
diff --git a/Omni/Bild/Deps/accelerate.nix b/Omni/Bild/Deps/accelerate.nix index be1d2fd..55a8609 100644 --- a/Omni/Bild/Deps/accelerate.nix +++ b/Omni/Bild/Deps/accelerate.nix @@ -1,10 +1,16 @@ -{ fetchFromGitHub, buildPythonPackage, numpy, packaging, psutil, pyyaml, torch +{ + fetchFromGitHub, + buildPythonPackage, + numpy, + packaging, + psutil, + pyyaml, + torch, }: - buildPythonPackage rec { name = "accelerate"; version = "0.15.0"; - propagatedBuildInputs = [ numpy packaging psutil pyyaml torch ]; + propagatedBuildInputs = [numpy packaging psutil pyyaml torch]; doCheck = false; src = fetchFromGitHub { owner = "huggingface"; @@ -13,4 +19,3 @@ buildPythonPackage rec { sha256 = "sha256-agfbOaa+Nm10HZkd2Y7zR3R37n+vLNsxCyxZax6O3Lo="; }; } - |