python311Packages.accelerate: fix build when torch.distributed is disabled

2023-11-05 12:44:28 +09:00
parent af016cfe8f
commit 42baf8f175
1 changed files with 10 additions and 0 deletions
@@ -2,6 +2,7 @@
 , lib
 , buildPythonPackage
 , fetchFromGitHub
+, fetchpatch
 , pythonAtLeast
 , pythonOlder
 , pytestCheckHook
@@ -29,6 +30,15 @@ buildPythonPackage rec {
    hash = "sha256-pFkEgE1NGLPBW1CeGU0RJr+1Nj/y58ZcljyOnJuR47A=";
  };

+  patches = [
+    # https://github.com/huggingface/accelerate/pull/2121
+    (fetchpatch {
+      name = "fix-import-error-without-torch_distributed.patch";
+      url = "https://github.com/huggingface/accelerate/commit/42048092eabd67a407ea513a62f2acde97079fbc.patch";
+      hash = "sha256-9lvnU6z5ZEFc5RVw2bP0cGVyrwAp/pxX4ZgnmCN7qH8=";
+    })
+  ];
+
  nativeBuildInputs = [ setuptools ];

  propagatedBuildInputs = [