diff --git a/src/py/flwr/cli/new/templates/app/pyproject.pytorch.toml.tpl b/src/py/flwr/cli/new/templates/app/pyproject.pytorch.toml.tpl index 04cdc5d94a18..81dd9fedc6b1 100644 --- a/src/py/flwr/cli/new/templates/app/pyproject.pytorch.toml.tpl +++ b/src/py/flwr/cli/new/templates/app/pyproject.pytorch.toml.tpl @@ -10,8 +10,8 @@ license = "Apache-2.0" dependencies = [ "flwr[simulation]>=1.14.0", "flwr-datasets[vision]>=0.3.0", - "torch==2.2.1", - "torchvision==0.17.1", + "torch==2.5.1", + "torchvision==0.20.1", ] [tool.hatch.build.targets.wheel]