diff --git a/pkgs/development/python-modules/accelerate/default.nix b/pkgs/development/python-modules/accelerate/default.nix index 811eb965dd0c..4fecc975b934 100644 --- a/pkgs/development/python-modules/accelerate/default.nix +++ b/pkgs/development/python-modules/accelerate/default.nix @@ -31,7 +31,7 @@ writableTmpDirAsHomeHook, }: -buildPythonPackage rec { +buildPythonPackage (finalAttrs: { pname = "accelerate"; version = "1.12.0"; pyproject = true; @@ -39,7 +39,7 @@ buildPythonPackage rec { src = fetchFromGitHub { owner = "huggingface"; repo = "accelerate"; - tag = "v${version}"; + tag = "v${finalAttrs.version}"; hash = "sha256-PwwaQSLOm+8Hd3trM1P+jRhYyoWM3QxOe5XT99haEmg="; }; @@ -108,6 +108,13 @@ buildPythonPackage rec { # TypeError: unsupported operand type(s) for /: 'NoneType' and 'int' (it seems cpuinfo doesn't work here) "test_mpi_multicpu_config_cmd" ] + ++ + lib.optionals + (stdenv.hostPlatform.isLinux && stdenv.hostPlatform.isAarch64 && (pythonAtLeast "3.14")) + [ + # RuntimeError: There is no current event loop in thread 'MainThread' + "test_accelerate_test" + ] ++ lib.optionals (!config.cudaSupport) [ # requires ptxas from cudatoolkit, which is unfree "test_dynamo_extract_model" @@ -187,9 +194,9 @@ buildPythonPackage rec { meta = { homepage = "https://huggingface.co/docs/accelerate"; description = "Simple way to train and use PyTorch models with multi-GPU, TPU, mixed-precision"; - changelog = "https://github.com/huggingface/accelerate/releases/tag/${src.tag}"; + changelog = "https://github.com/huggingface/accelerate/releases/tag/${finalAttrs.src.tag}"; license = lib.licenses.asl20; maintainers = with lib.maintainers; [ bcdarwin ]; mainProgram = "accelerate"; }; -} +})