Skip to content

Commit 45e9cf9

Browse files
authored
Update pyproject.toml
Pin PyTorch version to 2.5.1 to resolve torch.load weights_only error Fixes compatibility issue where PyTorch 2.6+ changed default weights_only behavior, causing existing weight files to fail loading with WeightsUnpickler errors.
1 parent 22172a8 commit 45e9cf9

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

pyproject.toml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -30,8 +30,8 @@ dependencies = [
3030
"PyYAML >= 5.3.1",
3131
"requests >= 2.23.0",
3232
"scipy >= 1.4.1, < 1.8.1",
33-
"torch >= 1.7.0, != 1.12.0",
34-
"torchvision >= 0.8.1, != 0.13.0",
33+
"torch == 2.5.1",
34+
"torchvision == 0.20.1",
3535
"tqdm >= 4.41.0",
3636
"protobuf < 4.21.3",
3737
"pandas >= 1.1.4",

0 commit comments

Comments
 (0)