Merge pull request #310829 from GaetanLepage/accelerate

python311Packages.accelerate: fix/skip failing darwin tests
This commit is contained in:
Pol Dellaiera
2024-05-11 23:05:37 +02:00
committed by GitHub
2 changed files with 9 additions and 0 deletions
@@ -4,6 +4,7 @@
, fetchFromGitHub
, pythonAtLeast
, pythonOlder
, llvmPackages
, pytest7CheckHook
, setuptools
, numpy
@@ -33,6 +34,10 @@ buildPythonPackage rec {
hash = "sha256-E20pI5BrcTrMYrhriuOUl5/liSaQQy6eqRyCoauwb9Q=";
};
buildInputs = [
llvmPackages.openmp
];
build-system = [ setuptools ];
dependencies = [
@@ -103,6 +108,8 @@ buildPythonPackage rec {
"accelerate"
];
__darwinAllowLocalNetworking = true;
meta = with lib; {
homepage = "https://huggingface.co/docs/accelerate";
description = "A simple way to train and use PyTorch models with multi-GPU, TPU, mixed-precision";
@@ -202,6 +202,8 @@ buildPythonPackage rec {
#"-W" "ignore" # uncomment for debugging help
];
__darwinAllowLocalNetworking = true;
# check the binary works outside the build env
doInstallCheck = true;
postInstallCheck = ''