From 67b9f15108486a1a29b348031e1b50a82fa55b40 Mon Sep 17 00:00:00 2001 From: init-22 Date: Fri, 15 Nov 2024 00:09:04 +0530 Subject: [PATCH] fix: pytorch version change --- setup.cfg | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/setup.cfg b/setup.cfg index e952513df..a74faa197 100644 --- a/setup.cfg +++ b/setup.cfg @@ -141,15 +141,15 @@ jax_gpu = # PyTorch CPU pytorch_cpu = - torch==2.5.0 - torchvision==0.20.0 + torch==2.5.1 + torchvision==0.20.1 # PyTorch GPU # Note: omit the cuda suffix and installing from the appropriate # wheel will result in using locally installed CUDA. pytorch_gpu = - torch==2.5.0 - torchvision==0.20.0 + torch==2.5.1 + torchvision==0.20.1 ###############################################################################