Upgraded torch version

PierreMarza · PierreMarza · commit 3e79d76ecb32 · 2025-10-17T11:48:04.000+02:00
diff --git a/setup.py b/setup.py
@@ -35,9 +35,9 @@
         "sentencepiece==0.2.1",
         "tabulate==0.9.0",
         "timm==1.0.13",
-        "torch==2.5.1",
-        "torchaudio==2.5.1",
-        "torchvision==0.20.1",
+        "torch==2.6.0",
+        "torchaudio==2.6.0",
+        "torchvision==0.21.0",
         "tqdm==4.66.5",
         "transformers==4.57.1",
         "wandb==0.19.4",
