fix pytorch version

priyakasimbeg · priyakasimbeg · commit 082be0311894 · 2025-02-08T02:27:45.000Z
diff --git a/pyproject.toml b/pyproject.toml
@@ -116,7 +116,7 @@ jax_gpu = [
 ]
 pytorch_cpu = ["torch==2.5.1", "torchvision==0.20.1"]
 pytorch_gpu = [
-  "torch==2.5.0",
+  "torch==2.5.1",
   "torchvision==0.20.1",
 ] # Note: omit the cuda suffix and installing from the appropriate wheel will result in using locally installed CUDA.
 wandb = ["wandb==0.16.5"]

Original file line number	Diff line number	Diff line change
`@@ -116,7 +116,7 @@ jax_gpu = [`
`116`	`116`	`]`
`117`	`117`	`pytorch_cpu = ["torch==2.5.1", "torchvision==0.20.1"]`
`118`	`118`	`pytorch_gpu = [`
`119`		`- "torch==2.5.0",`
	`119`	`+ "torch==2.5.1",`
`120`	`120`	`"torchvision==0.20.1",`
`121`	`121`	`] # Note: omit the cuda suffix and installing from the appropriate wheel will result in using locally installed CUDA.`
`122`	`122`	`wandb = ["wandb==0.16.5"]`