Commit 654f69d2 authored by Gaetan Lepage's avatar Gaetan Lepage
Browse files

python311Packages.accelerate: fix darwin tests

parent f8210014

pkgs/development/python-modules/accelerate/default.nix

+7 −0
Original line number Diff line number Diff line
@@ -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";