talexander/nixpkgs
1
0
Fork 0
Code Issues Pull Requests Releases Activity

python3Packages.torchsnapshot: fix torch.load call (#423123)

Browse Source
This commit is contained in:
Gaétan Lepage 2025-07-07 10:24:17 +02:00 committed by GitHub
commit 0826f5fcff
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 10 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
pkgs/development/python-modules/torchsnapshot/default.nix
View File

@ -33,6 +33,16 @@ buildPythonPackage rec {
hash = "sha256-F8OaxLH8BL6MPNLFv1hBuVmeEdnEQ5w2Qny6by1wP6k="; hash = "sha256-F8OaxLH8BL6MPNLFv1hBuVmeEdnEQ5w2Qny6by1wP6k=";
}; };
# _pickle.UnpicklingError: Weights only load failed.
# torchsnapshot needs to adapt to the change of torch.load that occured in 2.6.0:
# https://pytorch.org/docs/stable/generated/torch.load.html
postPatch = ''
substituteInPlace torchsnapshot/io_preparers/object.py \
--replace-fail \
"torch.load(io.BytesIO(buf))" \
"torch.load(io.BytesIO(buf), weights_only=False)"
'';
build-system = [ build-system = [
setuptools setuptools
]; ];