Unverified Commit 9718b32d authored by Martin Weinelt's avatar Martin Weinelt
Browse files

python3Packages.accelerate: 0.27.0 -> 0.29.3

parent 57cf17d3
Loading
Loading
Loading
Loading
+7 −2
Original line number Diff line number Diff line
@@ -2,6 +2,7 @@
, lib
, buildPythonPackage
, fetchFromGitHub
, pythonAtLeast
, pythonOlder
, pytest7CheckHook
, setuptools
@@ -20,7 +21,7 @@

buildPythonPackage rec {
  pname = "accelerate";
  version = "0.27.0";
  version = "0.29.3";
  pyproject = true;

  disabled = pythonOlder "3.7";
@@ -29,7 +30,7 @@ buildPythonPackage rec {
    owner = "huggingface";
    repo = pname;
    rev = "refs/tags/v${version}";
    hash = "sha256-7rnI8UXyAql8fLMKoSRrWzVw5CnyYVE2o6dJOzSgWxw=";
    hash = "sha256-oQGb/hlMN8JfwEyWufBvMk2Z1FMSl1lsdIbgZ3ZMdF8=";
  };

  nativeBuildInputs = [ setuptools ];
@@ -73,6 +74,10 @@ buildPythonPackage rec {

    # set the environment variable, CC, which conflicts with standard environment
    "test_patch_environment_key_exists"
  ] ++ lib.optionals (pythonAtLeast "3.12") [
    # RuntimeError: Dynamo is not supported on Python 3.12+
    "test_convert_to_fp32"
    "test_send_to_device_compiles"
  ] ++ lib.optionals (stdenv.isLinux && stdenv.isAarch64) [
    # usual aarch64-linux RuntimeError: DataLoader worker (pid(s) <...>) exited unexpectedly
    "CheckpointTest"