summaryrefslogtreecommitdiff
path: root/Omni/Bild/Deps/accelerate.nix
diff options
context:
space:
mode:
Diffstat (limited to 'Omni/Bild/Deps/accelerate.nix')
-rw-r--r--Omni/Bild/Deps/accelerate.nix13
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=";
};
}
-