diff --git a/project/pyproject.toml.jinja b/project/pyproject.toml.jinja index 54fe93e..b38ccbd 100644 --- a/project/pyproject.toml.jinja +++ b/project/pyproject.toml.jinja @@ -21,8 +21,8 @@ keywords = [] [tool.poetry.dependencies] python = "^{{ python_version }}" -{% if include_sample_code %}torch = "^2.3.1" -lightning = "^2.3.3"{% endif %} +{% if include_sample_code %}torch = "^2.4.0" +lightning = "^2.4.0"{% endif %} [tool.poetry.group.dev.dependencies] ruff = "^0.5.7"