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

Merge pull request #228748 from bcdarwin/unbreak-accelerate

Browse source 
python310Packages.accelerate: unbreak
This commit is contained in:
Weijia Wang 2023-04-30 01:05:30 +03:00 committed by GitHub
commit 430067be99
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
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:
"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"