diff --git a/pyproject.toml b/pyproject.toml index e13b402..5296eaf 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -25,7 +25,7 @@ packages = [ [tool.poetry.dependencies] python = "~3.10" -torch = "~=2.8.0" +torch = "~=2.10.0" torchvision = ">=0.16.0" lightning = ">=2.0.0" torchmetrics = ">=0.11.4"