Commit c7383a8e authored by Ben Darwin's avatar Ben Darwin
Browse files

python310Packages.accelerate: unbreak 

Disable known failing test on torch>=2.0
parent 5f40a2b9

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"