Unverified Commit 430067be authored by Weijia Wang's avatar Weijia Wang Committed by GitHub
Browse files

Merge pull request #228748 from bcdarwin/unbreak-accelerate 

python310Packages.accelerate: unbreak
parents 874bb5fc c7383a8e

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

+3 −0
Original line number Diff line number Diff line
@@ -53,6 +53,9 @@ buildPythonPackage rec { 
    # try to download data:

    "FeatureExamplesTests"

    "test_infer_auto_device_map_on_t0pp"

    # known failure with Torch>2.0; see https://github.com/huggingface/accelerate/pull/1339:

    # (remove for next release)

    "test_gradient_sync_cpu_multi"

  ] ++ lib.optionals (stdenv.isLinux && stdenv.isAarch64) [

    # usual aarch64-linux RuntimeError: DataLoader worker (pid(s) <...>) exited unexpectedly

    "CheckpointTest"