diff options
author | Gaetan Lepage <gaetan@glepage.com> | 2024-05-11 13:55:22 +0200 |
---|---|---|
committer | Gaetan Lepage <gaetan@glepage.com> | 2024-05-11 18:55:42 +0200 |
commit | 654f69d29f0d78294eaaf77923a7f83e17010990 (patch) | |
tree | 9ab8cc534f9c58dd4482dbfccb7da3be2101b476 | |
parent | f8210014c03dbec018d66b3e8d4125900b754196 (diff) |
python311Packages.accelerate: fix darwin tests
-rw-r--r-- | pkgs/development/python-modules/accelerate/default.nix | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/pkgs/development/python-modules/accelerate/default.nix b/pkgs/development/python-modules/accelerate/default.nix index 08874ce2e0fc2..61b3c53e834cb 100644 --- a/pkgs/development/python-modules/accelerate/default.nix +++ b/pkgs/development/python-modules/accelerate/default.nix @@ -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"; |