diff options
Diffstat (limited to 'Biz/Bild/Deps/accelerate.nix')
-rw-r--r-- | Biz/Bild/Deps/accelerate.nix | 24 |
1 files changed, 24 insertions, 0 deletions
diff --git a/Biz/Bild/Deps/accelerate.nix b/Biz/Bild/Deps/accelerate.nix new file mode 100644 index 0000000..5d00f14 --- /dev/null +++ b/Biz/Bild/Deps/accelerate.nix @@ -0,0 +1,24 @@ +{ fetchFromGitHub +, buildPythonPackage +, numpy +, packaging +, psutil +, pyyaml +, torch +}: + +buildPythonPackage rec { + name = "accelerate"; + version = "0.15.0"; + propagatedBuildInputs = [ + numpy packaging psutil pyyaml torch + ]; + doCheck = false; + src = fetchFromGitHub { + owner = "huggingface"; + repo = "accelerate"; + rev = "v${version}"; + sha256 = "sha256-agfbOaa+Nm10HZkd2Y7zR3R37n+vLNsxCyxZax6O3Lo="; + }; +} + |