python311Packages.accelerate: fix darwin tests
This commit is contained in:
@@ -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";
|
||||
|
||||
Reference in New Issue
Block a user