diff --git a/pyproject.toml b/pyproject.toml index 6440af7b..c2b35187 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -11,7 +11,7 @@ authors = [ requires-python = ">=3.9" dependencies = [ - "torch ~= 1.13.1", + "torch ~= 2.0.0", "torchvision ~= 0.14.1", "pytorch-lightning ~= 1.6.5", "torchmetrics == 0.6.0",