diff --git a/pyproject.toml b/pyproject.toml index 6a80d751..9015ac05 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -11,8 +11,8 @@ authors = [ requires-python = ">=3.9" dependencies = [ - "torch ~= 2.0.1", - "torchvision ~= 0.15.2", + "torch ~= 2.0.0", # Allow 2.0.0, because Poetry gets incomplete dependency for 2.0.1: https://github.com/python-poetry/poetry/issues/7902 + "torchvision ~= 0.15", "lightning[extra] ~= 2.0.5", # Full functionality including TensorboardX. "pydantic == 1.10.11", # https://github.com/Lightning-AI/lightning/pull/18022/files "torchmetrics == 1.0.0",