cafkafk/nixpkgs
cafkafk/nixpkgs
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 2

python310Packages.accelerate: unbreak

Browse source 
Disable known failing test on torch>=2.0
This commit is contained in:
Ben Darwin 2023-04-28 11:40:53 -04:00
parent 5f40a2b935
commit c7383a8e78
1 changed files with 3 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
pkgs/development/python-modules/accelerate/default.nix
View file

@ -53,6 +53,9 @@ buildPythonPackage rec {
# try to download data: # try to download data:
"FeatureExamplesTests" "FeatureExamplesTests"
"test_infer_auto_device_map_on_t0pp" "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) [ ] ++ lib.optionals (stdenv.isLinux && stdenv.isAarch64) [
# usual aarch64-linux RuntimeError: DataLoader worker (pid(s) <...>) exited unexpectedly # usual aarch64-linux RuntimeError: DataLoader worker (pid(s) <...>) exited unexpectedly
"CheckpointTest" "CheckpointTest"