From bf35e758d4b17d0cbc1686d09fed83baef41c717 Mon Sep 17 00:00:00 2001 From: Charles Beauville Date: Fri, 20 Dec 2024 16:19:16 +0100 Subject: [PATCH] ci(framework) Bump torch version --- src/py/flwr/cli/new/templates/app/pyproject.pytorch.toml.tpl | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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]