overrides: fix up torch 2.1.2

This commit is contained in:
Yang, Bo 2024-05-06 06:13:51 +00:00
parent 6677bc70a0
commit a00527b969
3 changed files with 459 additions and 493 deletions

View file

@ -3532,6 +3532,7 @@ lib.composeManyExtensions [
propagatedBuildInputs = (old.propagatedBuildInputs or [ ]) ++ [
final.numpy
final.packaging
];
});

File diff suppressed because it is too large Load diff

View file

@ -8,8 +8,8 @@ packages = [{include = "ml_stack"}]
[tool.poetry.dependencies]
python = "^3.10"
torch = "2.0.0"
torchvision = "0.15.1"
torch = "^2.1.2"
torchvision = "^0.16"
[build-system]
requires = ["poetry-core"]