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