blob: be1d2fd6c28c3095fa7364e2d332cf072065db05 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
|
{ 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=";
};
}
|
