update pytorch package

priyakasimbeg · priyakasimbeg · commit 400640a4553d · 2026-01-29T03:30:17.000Z
diff --git a/pyproject.toml b/pyproject.toml
@@ -108,8 +108,8 @@ jax_gpu = [
 ]
 
 pytorch_cpu = [
-  "torch==2.5.1", 
-  "torchvision==0.20.1"
+  "torch==2.9.0", 
+  "torchvision==0.24.0"
 ]
 pytorch_gpu = [
   "torch==2.9.0",
