Biz/Bild/Deps/accelerate.nix


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
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=";
  };
}